diff options
Diffstat (limited to 'pkgs/development/python-modules/skorch/default.nix')
-rw-r--r-- | pkgs/development/python-modules/skorch/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/skorch/default.nix b/pkgs/development/python-modules/skorch/default.nix index 0b2056979e4..a6508ded491 100644 --- a/pkgs/development/python-modules/skorch/default.nix +++ b/pkgs/development/python-modules/skorch/default.nix @@ -7,7 +7,7 @@ , flaky , numpy , pandas -, pytorch +, torch , scikit-learn , scipy , tabulate @@ -23,7 +23,7 @@ buildPythonPackage rec { sha256 = "b35cb4e50045742f0ffcfad33044af691d5d36b50212573753a804483a947ca9"; }; - propagatedBuildInputs = [ numpy pytorch scikit-learn scipy tabulate tqdm ]; + propagatedBuildInputs = [ numpy torch scikit-learn scipy tabulate tqdm ]; checkInputs = [ pytest pytest-cov flaky pandas pytestCheckHook ]; disabledTests = [ |