summary refs log tree commit diff
path: root/pkgs/development/python-modules/ignite
diff options
context:
space:
mode:
authorBen Darwin <bcdarwin@gmail.com>2021-11-29 20:00:16 -0500
committerBen Darwin <bcdarwin@gmail.com>2021-11-29 20:00:16 -0500
commit6f968f15727cbf39adbb0d60623f2c11f56eab6b (patch)
tree2f614e8eaaa171277cbd88c0208a218256ee7151 /pkgs/development/python-modules/ignite
parent078f5813f234bd4ee5f540fe025f89a932f7e454 (diff)
downloadnixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.tar
nixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.tar.gz
nixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.tar.bz2
nixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.tar.lz
nixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.tar.xz
nixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.tar.zst
nixpkgs-6f968f15727cbf39adbb0d60623f2c11f56eab6b.zip
python3Packages.ignite: remove optional pynvml dependency
Diffstat (limited to 'pkgs/development/python-modules/ignite')
-rw-r--r--pkgs/development/python-modules/ignite/default.nix4
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/ignite/default.nix b/pkgs/development/python-modules/ignite/default.nix
index 177fd891560..da5dfddcfb1 100644
--- a/pkgs/development/python-modules/ignite/default.nix
+++ b/pkgs/development/python-modules/ignite/default.nix
@@ -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