17 buildPythonPackage rec {
22 disabled = pythonOlder "3.8";
24 src = fetchFromGitHub {
28 hash = "sha256-J0xrqAGwH0bAs59T7zA8irMWOGbE2+Zd9kwqxYUYYMA=";
31 build-system = [ setuptools ];
46 # runs successfully in 3.9, however, async isn't correctly closed so it will fail after test suite.
47 doCheck = pythonOlder "3.9";
49 # Some packages are not in NixPkgs; other tests try to build distributed
50 # models, which doesn't work in the sandbox.
51 # avoid tests which need special packages
53 "--ignore=tests/ignite/contrib/handlers/test_clearml_logger.py"
54 "--ignore=tests/ignite/contrib/handlers/test_lr_finder.py"
55 "--ignore=tests/ignite/contrib/handlers/test_trains_logger.py"
56 "--ignore=tests/ignite/metrics/nlp/test_bleu.py"
57 "--ignore=tests/ignite/metrics/nlp/test_rouge.py"
58 "--ignore=tests/ignite/metrics/gan" # requires pytorch_fid; tries to download model to $HOME
59 "--ignore=tests/ignite/metrics/test_dill.py"
60 "--ignore=tests/ignite/metrics/test_psnr.py"
61 "--ignore=tests/ignite/metrics/test_ssim.py"
65 # disable tests which need specific packages
70 "test_gpu_info" # needs pynvml
72 "test_output_handler" # needs mlflow
73 "test_pbar" # slight output differences
74 "test_setup_clearml_logging"
82 pythonImportsCheck = [
94 description = "High-level training library for PyTorch";
95 homepage = "https://pytorch-ignite.ai";
96 changelog = "https://github.com/pytorch/ignite/releases/tag/v${version}";
97 license = lib.licenses.bsd3;
98 maintainers = [ lib.maintainers.bcdarwin ];
99 # ModuleNotFoundError: No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package
100 broken = stdenv.hostPlatform.isDarwin;