install mpi4py in mpi dockerfile

2018-10-31 11:34:10 -07:00
parent b4a149a75f
commit 58801032fc
1 changed files with 1 additions and 1 deletions
--- 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