diff --git a/pkgs/development/python-modules/imbalanced-learn/default.nix b/pkgs/development/python-modules/imbalanced-learn/default.nix index 794715cb4815..15ac55cb6783 100644 --- a/pkgs/development/python-modules/imbalanced-learn/default.nix +++ b/pkgs/development/python-modules/imbalanced-learn/default.nix @@ -14,6 +14,7 @@ threadpoolctl, pytestCheckHook, python, + sklearn-compat, }: buildPythonPackage rec { @@ -39,6 +40,7 @@ buildPythonPackage rec { scikit-learn scipy threadpoolctl + sklearn-compat ]; optional-dependencies = { @@ -74,12 +76,6 @@ buildPythonPackage rec { "imblearn/utils/tests/test_min_dependencies.py" ]; - disabledTests = [ - # Broken upstream test https://github.com/scikit-learn-contrib/imbalanced-learn/issues/1131 - "test_estimators_compatibility_sklearn" - "test_balanced_bagging_classifier_with_function_sampler" - ]; - meta = { 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";