[tune] Add more intensive tests (#7667)

* make_heavier_tests

* help
This commit is contained in:
Richard Liaw
2020-04-20 11:14:44 -07:00
committed by GitHub
parent 793e616a2d
commit 9f3e9e7e9f
6 changed files with 12 additions and 27 deletions
+2 -3
View File
@@ -9,10 +9,9 @@ RUN apt-get install -y zlib1g-dev libgl1-mesa-dev
RUN pip install -U pip
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
RUN pip install --upgrade bayesian-optimization hyperopt
RUN pip install ConfigSpace==0.4.10
RUN pip install --upgrade sigopt nevergrad scikit-optimize hpbandster lightgbm xgboost torch torchvision tensorboardX dragonfly-opt zoopt
RUN pip install -U tabulate mlflow
RUN pip install -U pytest-remotedata>=0.3.1
RUN pip install -U matplotlib jupyter pandas
RUN pip install -U matplotlib jupyter pandas
+1 -1
View File
@@ -13,7 +13,7 @@ RUN pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pyto
RUN pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.1.0-cp36-cp36m-manylinux2010_x86_64.whl
COPY requirements.txt .
RUN pip install -r requirements.txt
RUN pip install -r requirements.txt
# We port the source code in so that we run the most up-to-date stress tests.
ADD ray.tar /ray