From 0b2792a530c0143adbe34e3884532b702ad94606 Mon Sep 17 00:00:00 2001
From: Joe Zuntz <joezuntz@googlemail.com>
Date: Tue, 23 Jul 2024 14:13:54 +0100
Subject: [PATCH] try mpi4py with pip instead of conda

---
 txpipe-dev/Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/txpipe-dev/Dockerfile b/txpipe-dev/Dockerfile
index 01a9d9e..2e5069c 100644
--- 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