diff options
author | Benjamin Hipple <bhipple@protonmail.com> | 2020-05-25 20:51:03 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-05-25 20:51:03 -0400 |
commit | cd9fb17b75920f75dbb0e9dc647a91033be98622 (patch) | |
tree | 3d4ee5363734ac67a4526b88ea84b5b4d7111743 | |
parent | aa364b90d4e3952b52de3a54078c1b6451677871 (diff) | |
parent | 710a48751a4db638626a2f6efd6c0f604e26cb7d (diff) |
Merge pull request #76997 from bcdarwin/skorch
python3Packages.skorch: init at 0.8.0
-rw-r--r-- | pkgs/development/python-modules/skorch/default.nix | 42 | ||||
-rw-r--r-- | pkgs/top-level/python-packages.nix | 2 |
2 files changed, 44 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/skorch/default.nix b/pkgs/development/python-modules/skorch/default.nix new file mode 100644 index 0000000000000..8e3fcc12f5525 --- /dev/null +++ b/pkgs/development/python-modules/skorch/default.nix @@ -0,0 +1,42 @@ +{ lib +, buildPythonPackage +, fetchPypi +, pytestCheckHook +, pytest +, pytestcov +, flaky +, numpy +, pandas +, pytorch +, scikitlearn +, scipy +, tabulate +, tqdm +}: + +buildPythonPackage rec { + pname = "skorch"; + version = "0.8.0"; + + src = fetchPypi { + inherit pname version; + sha256 = "1l576dws9drjakfsn0pfpbr48b21vpxv3vd3dz8lkbn8q71zs22r"; + }; + + propagatedBuildInputs = [ numpy pytorch scikitlearn scipy tabulate tqdm ]; + checkInputs = [ pytest pytestcov flaky pandas pytestCheckHook ]; + + # on CPU, these expect artifacts from previous GPU run + disabledTests = [ + "test_load_cuda_params_to_cpu" + "test_pickle_load" + ]; + + meta = with lib; { + description = "Scikit-learn compatible neural net library using Pytorch"; + homepage = "https://skorch.readthedocs.io"; + changelog = "https://github.com/skorch-dev/skorch/blob/master/CHANGES.md"; + license = licenses.bsd3; + maintainers = with maintainers; [ bcdarwin ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 138b92c5ab5c6..4fd9cd32ed3a9 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1505,6 +1505,8 @@ in { sklearn-deap = callPackage ../development/python-modules/sklearn-deap { }; + skorch = callPackage ../development/python-modules/skorch { }; + slackclient = callPackage ../development/python-modules/slackclient { }; slicedimage = callPackage ../development/python-modules/slicedimage { }; |