fix scikit-learn deprecation issue

oracle · Dec 10, 2024 · 6490395 · 6490395
1 parent 013337d
commit 6490395
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -71,7 +71,7 @@ dependencies = [
   "psutil>=5.7.2",
   "python_jsonschema_objects>=0.3.13",
   "requests",
-  "scikit-learn>=1.0",
+  "scikit-learn>=1.0,<1.6.0",
   "tabulate>=0.8.9",
   "tqdm>=4.59.0",
   "pydantic>=2.6.3",
@@ -179,7 +179,7 @@ anomaly  = [
   "oracledb",
   "report-creator==1.0.28",
   "rrcf==0.4.4",
-  "scikit-learn",
+  "scikit-learn<1.6.0",
   "salesforce-merlion[all]==2.0.4"
 ]
 recommender = [