diff --git a/pkgs/development/python-modules/ignite/default.nix b/pkgs/development/python-modules/ignite/default.nix index d1bac885264f..efa9a412a05d 100644 --- a/pkgs/development/python-modules/ignite/default.nix +++ b/pkgs/development/python-modules/ignite/default.nix @@ -2,22 +2,23 @@ lib, buildPythonPackage, fetchFromGitHub, + pythonOlder, + setuptools, pytestCheckHook, pytest-xdist, torchvision, - pythonOlder, matplotlib, mock, packaging, torch, - scikit-learn, - tqdm, }: buildPythonPackage rec { pname = "ignite"; version = "0.5.1"; - format = "setuptools"; + pyproject = true; + + disabled = pythonOlder "3.8"; src = fetchFromGitHub { owner = "pytorch"; @@ -26,6 +27,13 @@ buildPythonPackage rec { hash = "sha256-J0xrqAGwH0bAs59T7zA8irMWOGbE2+Zd9kwqxYUYYMA="; }; + build-system = [ setuptools ]; + + dependencies = [ + packaging + torch + ]; + nativeCheckInputs = [ pytestCheckHook matplotlib @@ -33,12 +41,6 @@ buildPythonPackage rec { pytest-xdist torchvision ]; - propagatedBuildInputs = [ - packaging - torch - scikit-learn - tqdm - ]; # runs successfully in 3.9, however, async isn't correctly closed so it will fail after test suite. doCheck = pythonOlder "3.9"; @@ -76,10 +78,22 @@ buildPythonPackage rec { "visdom" ]; - meta = with lib; { + pythonImportsCheck = [ + "ignite" + "ignite.engine" + "ignite.handlers" + "ignite.metrics" + "ignite.distributed" + "ignite.exceptions" + "ignite.utils" + "ignite.contrib" + ]; + + meta = { description = "High-level training library for PyTorch"; - homepage = "https://pytorch.org/ignite"; - license = licenses.bsd3; - maintainers = [ maintainers.bcdarwin ]; + homepage = "https://pytorch-ignite.ai"; + changelog = "https://github.com/pytorch/ignite/releases/tag/v${version}"; + license = lib.licenses.bsd3; + maintainers = [ lib.maintainers.bcdarwin ]; }; }