diff options
-rw-r--r-- | pkgs/development/python-modules/imbalanced-learn/0.4.nix | 29 | ||||
-rw-r--r-- | pkgs/top-level/python-packages.nix | 6 |
2 files changed, 34 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/imbalanced-learn/0.4.nix b/pkgs/development/python-modules/imbalanced-learn/0.4.nix new file mode 100644 index 000000000000..e7d2c2f37148 --- /dev/null +++ b/pkgs/development/python-modules/imbalanced-learn/0.4.nix @@ -0,0 +1,29 @@ +{ stdenv, buildPythonPackage, fetchPypi, scikitlearn, pandas, nose, pytest }: + +buildPythonPackage rec { + pname = "imbalanced-learn"; + version = "0.4.3"; + + src = fetchPypi { + inherit pname version; + sha256 = "5bd9e86e40ce4001a57426541d7c79b18143cbd181e3330c1a3e5c5c43287083"; + }; + + propagatedBuildInputs = [ scikitlearn ]; + checkInputs = [ nose pytest pandas ]; + checkPhase = '' + export HOME=$PWD + # skip some tests that fail because of minimal rounding errors + # or large dependencies + py.test imblearn -k 'not classification \ + and not _generator \ + and not _forest \ + and not wrong_memory' + ''; + + meta = with stdenv.lib; { + description = "Library offering a number of re-sampling techniques commonly used in datasets showing strong between-class imbalance"; + homepage = https://github.com/scikit-learn-contrib/imbalanced-learn; + license = licenses.mit; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index f3e774b91e36..be6a291b7c70 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -2353,7 +2353,11 @@ in { image-match = callPackage ../development/python-modules/image-match { }; - imbalanced-learn = callPackage ../development/python-modules/imbalanced-learn { }; + imbalanced-learn = + if isPy27 then + callPackage ../development/python-modules/imbalanced-learn/0.4.nix { } + else + callPackage ../development/python-modules/imbalanced-learn { }; immutables = callPackage ../development/python-modules/immutables {}; |