From efbb8ac7bba4d8302e1601f7a3e26137e98b4133 Mon Sep 17 00:00:00 2001 From: Shaojun Liu <61072813+liu-shaojun@users.noreply.github.com> Date: Wed, 19 Jun 2024 11:17:10 +0800 Subject: [PATCH] unpin scikit-learn version (#5177) --- python/chronos/src/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/chronos/src/setup.py b/python/chronos/src/setup.py index 0effd210db0..f9e2135f9ae 100644 --- a/python/chronos/src/setup.py +++ b/python/chronos/src/setup.py @@ -71,7 +71,7 @@ def setup_package(): license='Apache License, Version 2.0', url='https://github.com/intel-analytics/BigDL', packages=get_bigdl_packages(), - install_requires=['pandas>=1.0.5, <=1.3.5', 'scikit-learn>=0.22.0, <=1.0.2', + install_requires=['pandas>=1.0.5, <=1.3.5', 'scikit-learn', 'bigdl-nano==' + VERSION], extras_require={'pytorch': ['bigdl-nano[pytorch]==' + VERSION], 'tensorflow': ['bigdl-nano=='+VERSION] + TF_DEP,