From bc637a2546d1fc6bfe1ee6da00de50ee4c19d0f9 Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Sun, 8 Mar 2020 08:16:00 +0100 Subject: [PATCH] [Tune Jenkins tests] Add dm_tree to docker. (#7500) * Fix. * Rollback. * Add dm_tree to docker examples and tune_test containers. --- docker/examples/Dockerfile | 2 +- docker/tune_test/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/examples/Dockerfile b/docker/examples/Dockerfile index 15d422ba5..64fd9afce 100644 --- a/docker/examples/Dockerfile +++ b/docker/examples/Dockerfile @@ -9,7 +9,7 @@ RUN apt-get install -y zlib1g-dev libgl1-mesa-dev # The following is needed to support TensorFlow 1.14 RUN conda remove -y --force wrapt RUN pip install -U pip -RUN pip install gym[atari] opencv-python-headless tensorflow lz4 pytest-timeout smart_open tensorflow_probability +RUN pip install gym[atari] opencv-python-headless tensorflow lz4 pytest-timeout smart_open tensorflow_probability dm_tree RUN pip install -U h5py # Mutes FutureWarnings RUN pip install --upgrade bayesian-optimization RUN pip install --upgrade hyperopt==0.1.2 diff --git a/docker/tune_test/Dockerfile b/docker/tune_test/Dockerfile index b0d3eaf4d..7efc08d05 100644 --- a/docker/tune_test/Dockerfile +++ b/docker/tune_test/Dockerfile @@ -13,7 +13,7 @@ RUN pip install -U boto3 RUN apt-get install -y zlib1g-dev libgl1-mesa-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 torch torchvision +RUN pip install gym[atari]==0.10.11 opencv-python-headless tensorflow lz4 keras pytest-timeout smart_open torch torchvision dm_tree RUN pip install --upgrade bayesian-optimization RUN pip install --upgrade hyperopt==0.1.2 RUN pip install ConfigSpace==0.4.10