diff options
author | Ben Darwin <bcdarwin@gmail.com> | 2020-01-05 15:22:36 -0500 |
---|---|---|
committer | Ben Darwin <bcdarwin@gmail.com> | 2020-05-25 20:31:11 -0400 |
commit | 710a48751a4db638626a2f6efd6c0f604e26cb7d (patch) | |
tree | 0af47ce935fc2ec11c79c46d1958aa9216bbcb98 /pkgs/development/python-modules/skorch/default.nix | |
parent | f59b843a3b23121d2a74f76783220fce2bf2c346 (diff) | |
download | nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.tar nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.tar.gz nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.tar.bz2 nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.tar.lz nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.tar.xz nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.tar.zst nixpkgs-710a48751a4db638626a2f6efd6c0f604e26cb7d.zip |
skorch: init at 0.8.0
Diffstat (limited to 'pkgs/development/python-modules/skorch/default.nix')
-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 00000000000..8e3fcc12f55 --- /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 ]; + }; +} |