From 76b31dd702c7f7181b5f023abd67039ba7f3294e Mon Sep 17 00:00:00 2001
From: Nikos Tsakiridis <tsakirin@auth.gr>
Date: Wed, 18 Dec 2024 10:22:39 +0200
Subject: [PATCH] :pushpin: pin sklearn to 1.5.2 to fix
 https://github.com/dmlc/xgboost/pull/11021 until it's released

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 00b50ee..5cedfff 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@
         "pandas",
         "numpy<2.0.0",
         "scipy",
-        "scikit-learn",
+        "scikit-learn==1.5.2",
         "matplotlib",
         "cubist>=0.1.0",
         "seaborn",