Add ML packages

CoffeaTeam · Feb 7, 2023 · 34e7c89 · 34e7c89
1 parent 42e23e2
commit 34e7c89
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/docker/Dockerfile.cc-analysis-ubuntu b/docker/Dockerfile.cc-analysis-ubuntu
@@ -104,6 +104,8 @@ RUN pip install --no-cache-dir \
     func-adl-uproot \
     dask_jobqueue==0.7.3 \
     tornado==6.2 \
+    dask-ml \
+    mlflow \
     aiostream
 
 # REMOVE ME AFTER TEST:

diff --git a/docker/Dockerfile.cc-ubuntu b/docker/Dockerfile.cc-ubuntu
@@ -63,6 +63,9 @@ RUN pip install --no-cache-dir \
     descartes==1.1.0 \
     # JSON processor
     jq \
+    # ML packages
+    dask-ml \
+    mlflow \
     # coffea casa jobqueue modules
     coffea_casa -U