python3Packages.ignite: remove optional pynvml dependency

This commit is contained in:
Ben Darwin 2021-11-29 20:00:16 -05:00
parent 078f5813f2
commit 6f968f1572

View File

@ -8,7 +8,6 @@
, matplotlib
, mock
, pytorch
, pynvml
, scikit-learn
, tqdm
}:
@ -25,7 +24,7 @@ buildPythonPackage rec {
};
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.
doCheck = pythonOlder "3.9";
@ -51,6 +50,7 @@ buildPythonPackage rec {
"idist"
"mlflow"
"tensorboard"
"test_gpu_info" # needs pynvml
"test_integration"
"test_output_handler" # needs mlflow
"test_pbar" # slight output differences