From 58801032fc767bc4676c3dea396b6733d2d2b2ad Mon Sep 17 00:00:00 2001
From: Peter Zhokhov <peterz@openai.com>
Date: Wed, 31 Oct 2018 11:34:10 -0700
Subject: [PATCH] install mpi4py in mpi dockerfile

---
 test.dockerfile.py36-mpi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.dockerfile.py36-mpi b/test.dockerfile.py36-mpi
index d56966b..7057cfa 100644
--- a/test.dockerfile.py36-mpi
+++ b/test.dockerfile.py36-mpi
@@ -10,7 +10,7 @@ WORKDIR $CODE_DIR/baselines
 RUN rm -rf __pycache__ && \
     find . -name "*.pyc" -delete && \
     pip install tensorflow && \
-    pip install -e .[test]
+    pip install -e .[test,mpi]
 
 
 CMD /bin/bash