Something went wrong. Try again.
lol
Something went wrong. Try again.
2.3 kB · 99 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100{ lib, buildPythonPackage, fetchFromGitHub, hatchling, pytestCheckHook, pytest-xdist, torchvision, matplotlib, mock, packaging, torch,}:
buildPythonPackage rec { pname = "ignite"; version = "0.5.4"; pyproject = true;
src = fetchFromGitHub { owner = "pytorch"; repo = "ignite"; tag = "v${version}"; hash = "sha256-taf1T5zSX436Bn2qLL8ba/fXyFtu51uOvYcWerpEQ1E="; };
build-system = [ hatchling ];
dependencies = [ packaging torch ];
nativeCheckInputs = [ pytestCheckHook matplotlib mock pytest-xdist torchvision ];
# async isn't correctly closed so it will fail after test suite. doCheck = false;
enabledTestPaths = [ "tests/" ];
# Some packages are not in NixPkgs; other tests try to build distributed # models, which doesn't work in the sandbox. # avoid tests which need special packages disabledTestPaths = [ "tests/ignite/contrib/handlers/test_clearml_logger.py" "tests/ignite/contrib/handlers/test_lr_finder.py" "tests/ignite/contrib/handlers/test_trains_logger.py" "tests/ignite/metrics/nlp/test_bleu.py" "tests/ignite/metrics/nlp/test_rouge.py" "tests/ignite/metrics/gan" # requires pytorch_fid; tries to download model to $HOME "tests/ignite/metrics/test_dill.py" "tests/ignite/metrics/test_psnr.py" "tests/ignite/metrics/test_ssim.py" ];
# disable tests which need specific packages disabledTests = [ "idist" "mlflow" "tensorboard" "test_gpu_info" # needs pynvml "test_integration" "test_output_handler" # needs mlflow "test_pbar" # slight output differences "test_setup_clearml_logging" "test_setup_neptune" "test_setup_plx" "test_write_results" "trains" "visdom" ];
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-ignite.ai"; changelog = "https://github.com/pytorch/ignite/releases/tag/${src.tag}"; license = lib.licenses.bsd3; maintainers = [ lib.maintainers.bcdarwin ]; };}