Skip to content

Commit

Permalink
[tune] Upgrade various tune search library dependencies (#36641)
Browse files Browse the repository at this point in the history
A number of dependencies are currently outdated. This PR upgrades these dependencies to their latest versions:

- optuna
- hyperopt
- dragonfly-opt
- bayesian-optimization
- timm
- xgboost (train/xgboost-ray dependency)

Signed-off-by: Kai Fricke <kai@anyscale.com>
  • Loading branch information
krfricke authored Jun 22, 2023
1 parent 8e8e0a4 commit c89f312
Showing 1 changed file with 8 additions and 13 deletions.
21 changes: 8 additions & 13 deletions python/requirements/ml/requirements_tune.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,10 @@
aim==3.16.1
ax-platform[mysql]==0.2.6; python_version < '3.8'
ax-platform[mysql]==0.3.2; python_version >= '3.8'
bayesian-optimization==1.2.0
bayesian-optimization==1.4.3
comet-ml==3.31.9
ConfigSpace==0.4.18
dragonfly-opt==0.1.6
ConfigSpace==0.7.1
dragonfly-opt==0.1.7
flaml==1.1.1
freezegun==1.1.0
# Requires decord which is unavailable for arm64
Expand All @@ -18,7 +18,7 @@ autorom[accept-rom-license]; platform_machine != "arm64"
h5py==3.7.0
hpbandster==0.7.4
HEBO==0.3.2
hyperopt==0.2.5
hyperopt==0.2.7
jupyterlab==3.6.1
lightgbm==3.3.5
matplotlib!=3.4.3
Expand All @@ -27,7 +27,7 @@ mlflow==2.4.1; python_version > '3.7'
# Unavailable for arm64 in more recent versions
mxnet==1.8.0.post0; platform_machine != "arm64"
nevergrad==0.4.3.post7
optuna==2.10.0
optuna==3.2.0
# For HEBO compatibility
pymoo==0.5.0
pytest-remotedata==0.3.2
Expand All @@ -38,15 +38,10 @@ s3fs==2023.1.0
shortuuid==1.0.1
scikit-optimize==0.9.0
sigopt==7.5.0
timm==0.4.5
timm==0.9.2
transformers==4.18.0; python_version <= '3.6'
transformers==4.19.1; python_version > '3.6'
wandb==0.13.4
xgboost==1.3.3
xgboost==1.6.2; python_version <= '3.7'
xgboost==1.7.6; python_version > '3.7'
zoopt==0.4.1

# We need to pin scipy because Dragonfly/Bayesopt tests will time out otherwise.
# This is likely because of a subdependency or API change. We may have to
# upgrade dragonfly/bayesopt in sync.
# Todo (krfricke): Upgrade
scipy==1.7.3

0 comments on commit c89f312

Please sign in to comment.