Fix tensorflow-1.14 installation in jenkins (#5007)

pcmoritz · ericl · commit 2e342ef71fd1 · 2019-06-21T11:04:40.000-07:00
diff --git a/docker/examples/Dockerfile b/docker/examples/Dockerfile
@@ -5,6 +5,8 @@ FROM ray-project/deploy
 # This updates numpy to 1.14 and mutes errors from other libraries
 RUN conda install -y numpy
 RUN apt-get install -y zlib1g-dev
+# The following is needed to support TensorFlow 1.14
+RUN conda remove -y --force wrapt
 RUN pip install gym[atari] opencv-python-headless tensorflow lz4 keras pytest-timeout smart_open
 RUN pip install -U h5py  # Mutes FutureWarnings
 RUN pip install --upgrade bayesian-optimization
diff --git a/docker/tune_test/Dockerfile b/docker/tune_test/Dockerfile
@@ -7,6 +7,8 @@ FROM ray-project/base-deps
 RUN pip install -U https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-0.8.0.dev1-cp27-cp27mu-manylinux1_x86_64.whl boto3
 # We install this after the latest wheels -- this should not override the latest wheels.
 RUN apt-get install -y zlib1g-dev
+# The following is needed to support TensorFlow 1.14
+RUN conda remove -y --force wrapt
 RUN pip install gym[atari]==0.10.11 opencv-python-headless tensorflow lz4 keras pytest-timeout smart_open
 RUN pip install --upgrade bayesian-optimization
 RUN pip install --upgrade git+git://github.com/hyperopt/hyperopt.git
