[Tune] Pin Tune Dependencies (#13027)

Co-authored-by: Ian <ian.rodney@gmail.com>
This commit is contained in:
Amog Kamsetty
2021-01-07 14:03:06 -08:00
committed by GitHub
parent f68922d043
commit 43f70faa25
6 changed files with 79 additions and 74 deletions
+47
View File
@@ -0,0 +1,47 @@
ax-platform==0.1.9; python_version < '3.7'
ax-platform==0.1.19; python_version >= '3.7'
bayesian-optimization==1.2.0
ConfigSpace==0.4.10
dragonfly-opt==0.1.6
gluoncv==0.9.0
gpy==1.9.9
gym[atari]==0.18.0
h5py==3.1.0
hpbandster==0.7.4
hyperopt==0.2.5
jupyter==1.0.0
keras==2.4.3
kubernetes==12.0.1
lightgbm==3.1.1
matplotlib==3.3.3
mlflow==1.13.0
mxnet==1.7.0.post1
nevergrad==0.4.2.post5
optuna==2.3.0
pytest-remotedata==0.3.2
pytorch-lightning-bolts==0.2.5
pytorch-lightning==1.0.3
scikit-learn==0.22.2
scikit-optimize==0.8.1
sigopt==5.7.0
smart_open==4.0.1
tensorflow-probability==0.11.1
timm==0.3.2
torch==1.7.0;sys_platform=="darwin"
torchvision==0.8.1;sys_platform=="darwin"
# On non-OSX machines only install CPU version of torch and torchvision
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.7.0+cpu;sys_platform!="darwin"
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.8.1+cpu;sys_platform!="darwin"
transformers==3.1
tune-sklearn==0.2.1
wandb==0.10.12
xgboost==1.3.0.post0
xgboost_ray==0.0.1
zoopt==0.4.1
-38
View File
@@ -1,38 +0,0 @@
ax-platform
bayesian-optimization
ConfigSpace==0.4.10
dragonfly-opt
gluoncv
gorilla # Need this because bug in mlflow. Should be fixed in v1.12.2
gym[atari]
GPy
h5py
hpbandster
hyperopt>=0.2.5
jupyter
keras
kubernetes
lightgbm
matplotlib
mlflow
mxnet
nevergrad
optuna
pytest-remotedata>=0.3.1
pytorch-lightning==1.0.3
pytorch-lightning-bolts
scikit-learn
scikit-optimize
sigopt
smart_open
tensorflow-probability
timm
torch>=1.6.0
torchvision>=0.6.0
# transformers
git+git://github.com/huggingface/transformers.git@bdcc4b78a27775d1ec8f3fd297cb679c257289db#transformers
git+git://github.com/ray-project/tune-sklearn@master#tune-sklearn
git+git://github.com/ray-project/xgboost_ray@master#xgboost_ray
wandb
xgboost
zoopt>=0.4.1