CI dockerfiles with and without mpi

.travis.yml

@@ -5,10 +5,11 @@ python:
 services:
     - docker
 
-install:
+env:
-    - pip install flake8
+  - DOCKER_SUFFIX=py36-nompi
-    - docker build . -t baselines-test
+  - DOCKER_SUFFIX=py36-mpi
 
+install: "" # so travis doesn't do pip install requirements.txt
 script:
-    - flake8 . --show-source --statistics
+  - flake8 . --show-source --statistics
-    - docker run baselines-test pytest -v .
+  - docker build -f test.dockerfile.${DOCKER_SUFFIX} -t baselines-test .

test.dockerfile.py36-mpi

@@ -1,6 +1,6 @@
 FROM python:3.6
 
-# RUN apt-get -y update && apt-get -y install git wget python-dev python3-dev libopenmpi-dev python-pip zlib1g-dev cmake python-opencv
+RUN apt-get -y update && apt-get -y install ffmpeg libopenmpi-dev
 ENV CODE_DIR /root/code
 
 COPY . $CODE_DIR/baselines

test.dockerfile.py36-nompi

@@ -0,0 +1,16 @@
+FROM python:3.6
+
+RUN apt-get -y update && apt-get -y install ffmpeg
+ENV CODE_DIR /root/code
+
+COPY . $CODE_DIR/baselines
+WORKDIR $CODE_DIR/baselines
+
+# Clean up pycache and pyc files
+RUN rm -rf __pycache__ && \
+    find . -name "*.pyc" -delete && \
+    pip install tensorflow && \
+    pip install -e .[test]
+
+
+CMD /bin/bash