pkgs/development/python-modules/skorch/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , pytestCheckHook
   5 , pytest
   6 , pytest-cov
   7 , flaky
   8 , numpy
   9 , pandas
  10 , torch
  11 , scikit-learn
  12 , scipy
  13 , tabulate
  14 , tqdm
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "skorch";
  19   version = "0.11.0";
  20
  21   src = fetchPypi {
  22     inherit pname version;
  23     sha256 = "b35cb4e50045742f0ffcfad33044af691d5d36b50212573753a804483a947ca9";
  24   };
  25
  26   propagatedBuildInputs = [ numpy torch scikit-learn scipy tabulate tqdm ];
  27   checkInputs = [ pytest pytest-cov flaky pandas pytestCheckHook ];
  28
  29   disabledTests = [
  30     # on CPU, these expect artifacts from previous GPU run
  31     "test_load_cuda_params_to_cpu"
  32     # failing tests
  33     "test_pickle_load"
  34     "test_grid_search_with_slds_"
  35     "test_grid_search_with_dict_works"
  36   ];
  37
  38   meta = with lib; {
  39     description = "Scikit-learn compatible neural net library using Pytorch";
  40     homepage = "https://skorch.readthedocs.io";
  41     changelog = "https://github.com/skorch-dev/skorch/blob/master/CHANGES.md";
  42     license = licenses.bsd3;
  43     maintainers = with maintainers; [ bcdarwin ];
  44     # TypeError: __init__() got an unexpected keyword argument 'iid'
  45     broken = true;
  46   };
  47 }