try mpi4py with pip instead of conda

LSSTDESC · Jul 23, 2024 · 0b2792a · 0b2792a
1 parent 77ce453
commit 0b2792a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/txpipe-dev/Dockerfile b/txpipe-dev/Dockerfile
@@ -48,7 +48,9 @@ RUN git clone --branch ${TX_TAG_FOR_DEPENDENCIES}  https://github.com/LSSTDESC/T
     && cd .. \
     && rm -rf TXPipe \
     && pip uninstall -y jaxlib \ 
-    && mamba install -y --force-reinstall -c conda-forge jaxlib=0.4.23
+    && mamba install -y --force-reinstall -c conda-forge jaxlib=0.4.23 \
+    && conda remove --force mpi4py \
+    && pip install mpi4py
 
 # ^^ conda insists on uninstalling and reinstalling an incorrect jaxlib version