pkgs/development/python-modules/imbalanced-learn/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , isPy27
   5 , pandas
   6 , pytestCheckHook
   7 , scikit-learn
   8 }:
   9
  10 buildPythonPackage rec {
  11   pname = "imbalanced-learn";
  12   version = "0.9.1";
  13   disabled = isPy27; # scikit-learn>=0.21 doesn't work on python2
  14
  15   src = fetchPypi {
  16     inherit pname version;
  17     sha256 = "sha256-ThT3rmB4tauEO3PDebKsULIHRGQV2cJDjIhdbLWv2WI=";
  18   };
  19
  20   propagatedBuildInputs = [ scikit-learn ];
  21   checkInputs = [ pytestCheckHook pandas ];
  22   preCheck = ''
  23     export HOME=$TMPDIR
  24   '';
  25   disabledTests = [
  26     "estimator"
  27     "classification"
  28     "_generator"
  29     "show_versions"
  30     "test_make_imbalanced_iris"
  31     "test_rusboost[SAMME.R]"
  32
  33     # https://github.com/scikit-learn-contrib/imbalanced-learn/issues/824
  34     "ValueDifferenceMetric"
  35   ];
  36
  37   meta = with lib; {
  38     description = "Library offering a number of re-sampling techniques commonly used in datasets showing strong between-class imbalance";
  39     homepage = "https://github.com/scikit-learn-contrib/imbalanced-learn";
  40     license = licenses.mit;
  41     maintainers = [ maintainers.rmcgibbo ];
  42   };
  43 }