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 /pkgs/development/python-modules | |
parent | aa364b90d4e3952b52de3a54078c1b6451677871 (diff) | |
parent | 710a48751a4db638626a2f6efd6c0f604e26cb7d (diff) | |
download | nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.tar nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.tar.gz nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.tar.bz2 nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.tar.lz nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.tar.xz nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.tar.zst nixlib-cd9fb17b75920f75dbb0e9dc647a91033be98622.zip |
Merge pull request #76997 from bcdarwin/skorch
python3Packages.skorch: init at 0.8.0
Diffstat (limited to 'pkgs/development/python-modules')
-rw-r--r-- | pkgs/development/python-modules/skorch/default.nix | 42 |
1 files changed, 42 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 000000000000..8e3fcc12f552 --- /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 ]; + }; +} |