Update Dockerfile

LSSTDESC · Mar 15, 2024 · 5f1b6be · 5f1b6be
1 parent f8c47e3
commit 5f1b6be
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/txpipe-dev/Dockerfile b/txpipe-dev/Dockerfile
@@ -49,8 +49,8 @@ RUN mamba install -c conda-forge -y scipy=1.8.0 \
                      pygraphviz \
                      compilers \
                      guppy3 \
-                     "jax=0.4.19" \
-                     "jaxlib=0.4.19=*cpu*" \
+                     "jax=0.4.14" \
+                     "jaxlib=0.4.14=*cpu*" \
                      "dm-tree=0.1.7" \
                      "cosmosis=3.4" \
                      threadpoolctl \
@@ -85,7 +85,9 @@ RUN pip install "ceci>=1.17" \
                 "pz-rail-pipelines==0.1.0" \
                 "pz-rail-pzflow==0.0.1" \ 
                 "pz-rail-sklearn==0.0.1" \
-                "pz-rail-som==0.0.2"
+                "pz-rail-som==0.0.2" \
+                "jax==0.4.19" \ 
+                "jaxlib==0.4.19"
 
 # Force a version of qp that is on pypi but not yet conda
 RUN mamba uninstall --force qp && pip install qp-prob==0.8.5