@@ -8,7 +8,6 @@
88, matplotlib
99, mock
1010, pytorch
11−, pynvml
1211, scikit-learn
1312, tqdm
1413}:
@@ -25,7 +24,7 @@ buildPythonPackage rec {
2524 };
2625
2726 checkInputs = [ pytestCheckHook matplotlib mock pytest-xdist torchvision ];
28− propagatedBuildInputs = [ pytorch scikit-learn tqdm pynvml ];
27+ propagatedBuildInputs = [ pytorch scikit-learn tqdm ];
2928
3029 # runs succesfully in 3.9, however, async isn't correctly closed so it will fail after test suite.
3130 doCheck = pythonOlder "3.9";
@@ -51,6 +50,7 @@ buildPythonPackage rec {
5150 "idist"
5251 "mlflow"
5352 "tensorboard"
53+ "test_gpu_info" # needs pynvml
5454 "test_integration"
5555 "test_output_handler" # needs mlflow
5656 "test_pbar" # slight output differences