mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-03 12:23:02 +00:00
python3Packages.ignite: remove optional pynvml dependency
This commit is contained in:
parent
078f5813f2
commit
6f968f1572
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user