diff --git a/Dockerfile b/Dockerfile
index 322eb4091518a4ac77708f01cf6feeb265612164..724afacc7c84c790f28124ce1e3cf00082b63a34 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,4 +14,4 @@ RUN mamba install -c conda-forge jax
 RUN mamba install -c conda-forge flax
 
 # Install pip packages
-RUN pip install rlax
+#RUN pip install rlax