From 1fb3d1299d783ba44b7535a24138046992a65219 Mon Sep 17 00:00:00 2001 From: nickotto Date: Tue, 7 Jan 2025 12:09:21 -0800 Subject: [PATCH] scikit-learn compat issue --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b1d194f8..28fbf0d8 100644 --- a/setup.py +++ b/setup.py @@ -30,7 +30,7 @@ def calculate_version(): zip_safe=True, install_requires=['numpy==1.26.4', 'scipy>=1.3.1', - 'scikit-learn>=1.3.0', + 'scikit-learn>=1.4.2,<1.6', 'update_checker>=0.16', 'tqdm>=4.36.1', 'stopit>=1.1.1',