try forcing pip

Author: joezuntz

txpipe/Dockerfile

@@ -3,7 +3,8 @@ MAINTAINER [email protected]
 
 # conda dependencies. As much as possible goes in here!
 # should we be pinning versions?
-RUN mamba install -c conda-forge -y scipy=1.8.0 \
+RUN mamba install -c conda-forge -y \
+                     scipy=1.8.0 \
                      matplotlib=3.5.1 \
                      psutil \
                      "numpy=1.23.*" \
@@ -61,7 +62,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.14" \
+                "jax=lib=0.4.14"
 
 # 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