Upgrade scikit-learn to 1.5.0 to fix dependabot issue

intel-analytics · Jun 18, 2024 · 22aa986 · 22aa986
1 parent 4d6c379
commit 22aa986
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/chronos/src/setup.py b/python/chronos/src/setup.py
@@ -54,7 +54,7 @@ def setup_package():
         url='https://github.com/intel-analytics/BigDL',
         packages=get_bigdl_packages(),
         install_requires=['bigdl-orca=='+VERSION, 'bigdl-nano[pytorch]',
-                          'pandas>=1.0.5, <1.3.0', 'scikit-learn', 'statsmodels'],
+                          'pandas>=1.0.5, <1.3.0', 'scikit-learn>=1.5.0', 'statsmodels'],
         extras_require={'all': ['bigdl-orca[automl]=='+VERSION]},
         dependency_links=['https://d3kbcqa49mib13.cloudfront.net/spark-2.0.0-bin-hadoop2.7.tgz'],
         include_package_data=True,