pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 setuptools,
6 torch,
7 pytestCheckHook,
8}:
9
10buildPythonPackage rec {
11 pname = "geotorch";
12 version = "0.3.0";
13 pyproject = true;
14
15 src = fetchFromGitHub {
16 owner = "lezcano";
17 repo = "geotorch";
18 tag = version;
19 hash = "sha256-kkn0PZzQRodXCeX3RcajVvrp1TrhSVgKYwyJGAMuvLM=";
20 };
21
22 build-system = [ setuptools ];
23
24 dependencies = [
25 torch
26 ];
27
28 pythonImportsCheck = [ "geotorch" ];
29
30 nativeCheckInputs = [
31 pytestCheckHook
32 ];
33
34 meta = {
35 description = "Constrained optimization toolkit for PyTorch";
36 homepage = "https://github.com/lezcano/geotorch";
37 changelog = "https://github.com/lezcano/geotorch/releases/tag/${version}";
38 license = lib.licenses.mit;
39 maintainers = with lib.maintainers; [ flokli ];
40 };
41}