Pin sktime and scikit learn dependency to strict version to avoid Val…

…ue errors
mindsdb · Dec 19, 2024 · e9dafb6 · e9dafb6
1 parent 2c7f1fa
commit e9dafb6
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 70 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,10 +24,10 @@ transformers = ">=4.34.0"
 optuna = ">=3.1.0,<4.0.0"
 scipy = ">=1.5.4"
 psutil = ">=5.7.0"
-scikit-learn = ">=1.5.0"
+scikit-learn = "==1.5.0"
 dataclasses_json = ">=0.5.4"
 dill = "==0.3.6"
-sktime = ">=0.30.0"
+sktime = "==0.35.0"
 statsforecast = "~=1.6.0"
 torch_optimizer = "==0.1.0"
 black = "==24.3.0"