diff options
-rw-r--r-- | pkgs/development/python-modules/grad-cam/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/grad-cam/default.nix b/pkgs/development/python-modules/grad-cam/default.nix index 4e43915c86dfa..bdfa8c0e49c0b 100644 --- a/pkgs/development/python-modules/grad-cam/default.nix +++ b/pkgs/development/python-modules/grad-cam/default.nix @@ -6,7 +6,7 @@ , numpy , opencv4 , pillow -, scikitlearn +, scikit-learn , torch , torchvision , ttach @@ -33,7 +33,7 @@ buildPythonPackage rec { numpy opencv4 pillow - scikitlearn + scikit-learn torchvision ttach tqdm |