diff options
Diffstat (limited to 'pkgs/development/python-modules/deepwave/default.nix')
-rw-r--r-- | pkgs/development/python-modules/deepwave/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/deepwave/default.nix b/pkgs/development/python-modules/deepwave/default.nix index 0e5e9b45661..f69c386047d 100644 --- a/pkgs/development/python-modules/deepwave/default.nix +++ b/pkgs/development/python-modules/deepwave/default.nix @@ -1,7 +1,7 @@ { lib , buildPythonPackage , fetchFromGitHub -, pytorch +, torch , ninja , scipy , which @@ -48,7 +48,7 @@ buildPythonPackage rec { export HOME=$(mktemp -d) ''; - propagatedBuildInputs = [ pytorch pybind11 ]; + propagatedBuildInputs = [ torch pybind11 ]; checkInputs = [ which |