diff --git a/setup.py b/setup.py index 915af01..50668b1 100644 --- a/setup.py +++ b/setup.py @@ -56,7 +56,7 @@ with open(path.join(here, "README.md"), "r", "utf-8") as f: extras = { "xgboost": ["xgboost>=0.90,<2"], - "scikit-learn": ["scikit-learn>=1.3,<1.4"], + "scikit-learn": ["scikit-learn>=1.3,<1.6"], "lightgbm": ["lightgbm>=2,<4"], "pytorch": [ "requests<3",