diff --git a/python/chronos/src/setup.py b/python/chronos/src/setup.py index b8f05987fb6..371560614a4 100644 --- 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,