TSML dependencies: Explicitly add dask-expr and distributed

amotl · amotl · commit 540547e85e6e · 2024-04-10T14:32:56.000+02:00
diff --git a/topic/machine-learning/automl/requirements.txt b/topic/machine-learning/automl/requirements.txt
@@ -1,6 +1,8 @@
 # Real.
 crate[sqlalchemy]
 dask>=2024.4.1  # Python 3.11.9 breaks previous Dask
+dask-expr<2
+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
diff --git a/topic/machine-learning/mlops-mlflow/requirements.txt b/topic/machine-learning/mlops-mlflow/requirements.txt
@@ -1,5 +1,7 @@
 # Real.
 dask>=2024.4.1  # Python 3.11.9 breaks previous Dask
+dask-expr<2
+distributed>=2024.4.1  # Python 3.11.9 breaks previous Dask
 mlflow-cratedb==2.11.3
 pydantic<3
 salesforce-merlion>=2,<3
diff --git a/topic/timeseries/requirements.txt b/topic/timeseries/requirements.txt
@@ -1,6 +1,8 @@
 crate[sqlalchemy]==0.35.2
 cratedb-toolkit[datasets]==0.0.8
 dask>=2024.4.1  # Python 3.11.9 breaks previous Dask
+dask-expr<2
+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
 refinitiv-data<1.7
 pandas<2
