TSML: Adjust dependencies around PyCaret/Joblib

Dependency hell caused a few hiccups. Subsequent bugfix releases help resolve the situation. In this case, PyCaret 3.3.1.
crate · Apr 16, 2024 · ae94c5c · ae94c5c
1 parent 8e45993
commit ae94c5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/topic/machine-learning/automl/requirements.txt b/topic/machine-learning/automl/requirements.txt
@@ -1,11 +1,8 @@
 # Real.
 crate[sqlalchemy]
-dask>=2024.4.1  # Python 3.11.9 breaks previous Dask
-distributed>=2024.4.1  # Python 3.11.9 breaks previous Dask
-joblib<1.4  # Joblib 1.4.0 is not compatible with PyCaret 3.3.0
 mlflow-cratedb==2.11.3
 plotly<5.21
-pycaret[models,parallel,test]==3.3.0
+pycaret[models,parallel,test]==3.3.1
 pydantic<2
 python-dotenv<2
 

diff --git a/topic/timeseries/requirements.txt b/topic/timeseries/requirements.txt
@@ -1,8 +1,7 @@
 crate[sqlalchemy]==0.35.2
 cratedb-toolkit[datasets]==0.0.10
-joblib<1.4  # Joblib 1.4.0 is not compatible with PyCaret 3.3.0
 refinitiv-data<1.7
 pandas<2
-pycaret>=3.0,<3.4
+pycaret==3.3.1
 pydantic<2
 sqlalchemy<2