diff --git a/pkgs/development/python-modules/ignite/default.nix b/pkgs/development/python-modules/ignite/default.nix new file mode 100644 index 000000000000..1ded86af3fb2 --- /dev/null +++ b/pkgs/development/python-modules/ignite/default.nix @@ -0,0 +1,38 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, pytest +, mock +, pytorch +, pynvml +, scikitlearn +, tqdm +}: + +buildPythonPackage rec { + pname = "ignite"; + version = "0.2.1"; + + src = fetchFromGitHub { + owner = "pytorch"; + repo = pname; + rev = "v${version}"; + sha256 = "15k6dd11yxn4923llcpmw4srl1by5ljhh7aw5pnkn4n4qpywh6cm"; + }; + + checkInputs = [ pytest mock ]; + + checkPhase = '' + pytest -k 'not visdom and not tensorboard and not mlflow and not polyaxon' tests/ + ''; + # these packages are not currently in nixpkgs + + propagatedBuildInputs = [ pytorch scikitlearn tqdm pynvml ]; + + meta = with lib; { + description = "High-level training library for PyTorch"; + homepage = https://pytorch.org/ignite; + license = licenses.bsd3; + maintainers = [ maintainers.bcdarwin ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 438eb0e43ad3..9658528b3b3d 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -761,6 +761,8 @@ in { i3ipc = callPackage ../development/python-modules/i3ipc { }; + ignite = callPackage ../development/python-modules/ignite { }; + ihatemoney = callPackage ../development/python-modules/ihatemoney { }; imutils = callPackage ../development/python-modules/imutils { };