Files
nixpkgs/pkgs/development/python-modules/geotorch/default.nix
2026-06-15 03:28:35 +00:00

42 lines
808 B
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
setuptools,
torch,
pytestCheckHook,
}:
buildPythonPackage rec {
pname = "geotorch";
version = "0.4.0";
pyproject = true;
src = fetchFromGitHub {
owner = "lezcano";
repo = "geotorch";
tag = version;
hash = "sha256-mMVgN8ZmedSz5VxAAE7vdvmZXiP5y3GkO60o5hCSHn8=";
};
build-system = [ setuptools ];
dependencies = [
torch
];
pythonImportsCheck = [ "geotorch" ];
nativeCheckInputs = [
pytestCheckHook
];
meta = {
description = "Constrained optimization toolkit for PyTorch";
homepage = "https://github.com/lezcano/geotorch";
changelog = "https://github.com/lezcano/geotorch/releases/tag/${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ flokli ];
};
}