diff options
Diffstat (limited to 'pkgs/development/python-modules/torchgpipe/default.nix')
-rw-r--r-- | pkgs/development/python-modules/torchgpipe/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/torchgpipe/default.nix b/pkgs/development/python-modules/torchgpipe/default.nix index 2c289f85269..68e113ce70f 100644 --- a/pkgs/development/python-modules/torchgpipe/default.nix +++ b/pkgs/development/python-modules/torchgpipe/default.nix @@ -4,7 +4,7 @@ , isPy27 , pytest-runner , pytestCheckHook -, pytorch +, torch }: buildPythonPackage rec { @@ -20,7 +20,7 @@ buildPythonPackage rec { sha256 = "0ki0njhmz1i3pkpr3y6h6ac7p5qh1kih06mknc2s18mfw34f2l55"; }; - propagatedBuildInputs = [ pytorch ]; + propagatedBuildInputs = [ torch ]; checkInputs = [ pytest-runner pytestCheckHook ]; disabledTests = [ |