diff options
Diffstat (limited to 'pkgs/development/python-modules/rising/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/rising/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/rising/default.nix b/pkgs/development/python-modules/rising/default.nix index b68fedbcfc0..07fdad78aad 100644 --- a/pkgs/development/python-modules/rising/default.nix +++ b/pkgs/development/python-modules/rising/default.nix @@ -6,7 +6,7 @@ , pytest-cov , dill , numpy -, pytorch +, torch , threadpoolctl , tqdm }: @@ -24,7 +24,7 @@ buildPythonPackage rec { sha256 = "15wYWToXRae1cMpHWbJwzAp0THx6ED9ixQgL+n1v9PI="; }; - propagatedBuildInputs = [ numpy pytorch threadpoolctl tqdm ]; + propagatedBuildInputs = [ numpy torch threadpoolctl tqdm ]; checkInputs = [ dill pytest-cov pytestCheckHook ]; disabledTests = [ "test_affine" ]; # deprecated division operator '/' |