diff options
Diffstat (limited to 'pkgs/development/python-modules/pywick/default.nix')
-rw-r--r-- | pkgs/development/python-modules/pywick/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/pywick/default.nix b/pkgs/development/python-modules/pywick/default.nix index 0db11576203..6cbb64845db 100644 --- a/pkgs/development/python-modules/pywick/default.nix +++ b/pkgs/development/python-modules/pywick/default.nix @@ -8,7 +8,7 @@ , pandas , pillow , six -, pytorch +, torch , torchvision , tqdm , lib @@ -28,7 +28,7 @@ buildPythonPackage rec { }; propagatedBuildInputs = [ - h5py hickle numpy pandas pillow six pytorch torchvision tqdm + h5py hickle numpy pandas pillow six torch torchvision tqdm ]; checkInputs = [ pytest ]; |