nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-12-04 21:03:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.ignite: remove optional pynvml dependency

Browse Source
This commit is contained in:
Ben Darwin 2021-11-29 20:00:16 -05:00
parent 078f5813f2
commit 6f968f1572
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pkgs/development/python-modules/ignite/default.nix
View File

@ -8,7 +8,6 @@
, matplotlib , matplotlib
, mock , mock
, pytorch , pytorch
, pynvml
, scikit-learn , scikit-learn
, tqdm , tqdm
}: }:
@ -25,7 +24,7 @@ buildPythonPackage rec {
}; };
checkInputs = [ pytestCheckHook matplotlib mock pytest-xdist torchvision ]; checkInputs = [ pytestCheckHook matplotlib mock pytest-xdist torchvision ];
propagatedBuildInputs = [ pytorch scikit-learn tqdm pynvml ]; propagatedBuildInputs = [ pytorch scikit-learn tqdm ];
# runs succesfully in 3.9, however, async isn't correctly closed so it will fail after test suite. # runs succesfully in 3.9, however, async isn't correctly closed so it will fail after test suite.
doCheck = pythonOlder "3.9"; doCheck = pythonOlder "3.9";
@ -51,6 +50,7 @@ buildPythonPackage rec {
"idist" "idist"
"mlflow" "mlflow"
"tensorboard" "tensorboard"
"test_gpu_info" # needs pynvml
"test_integration" "test_integration"
"test_output_handler" # needs mlflow "test_output_handler" # needs mlflow
"test_pbar" # slight output differences "test_pbar" # slight output differences