From 2bd36b256a0bfdf8f38b5c40b06ba31fbf98ffe8 Mon Sep 17 00:00:00 2001
From: Philip Wilk
Date: Sun, 15 Mar 2026 14:57:55 +0000
Subject: [PATCH 1/2] python3Packages.imbalanced-learn: re-add sklearn-compat
---
pkgs/development/python-modules/imbalanced-learn/default.nix | 2 ++
1 file changed, 2 insertions(+)
diff --git a/pkgs/development/python-modules/imbalanced-learn/default.nix b/pkgs/development/python-modules/imbalanced-learn/default.nix
index 794715cb4815..e22d0de6f104 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 = {
From f35639db1f965e2f5a1a67aaf695a4d1a10944ba Mon Sep 17 00:00:00 2001
From: Philip Wilk
Date: Sun, 15 Mar 2026 14:58:13 +0000
Subject: [PATCH 2/2] python3Packages.imbalanced-learn: enable tests fixed
upstream
---
.../development/python-modules/imbalanced-learn/default.nix | 6 ------
1 file changed, 6 deletions(-)
diff --git a/pkgs/development/python-modules/imbalanced-learn/default.nix b/pkgs/development/python-modules/imbalanced-learn/default.nix
index e22d0de6f104..15ac55cb6783 100644
--- a/pkgs/development/python-modules/imbalanced-learn/default.nix
+++ b/pkgs/development/python-modules/imbalanced-learn/default.nix
@@ -76,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";