diff --git a/python/requirements/tune/requirements_tune.txt b/python/requirements/tune/requirements_tune.txt index c70765c8af9..003fdda48ff 100644 --- a/python/requirements/tune/requirements_tune.txt +++ b/python/requirements/tune/requirements_tune.txt @@ -14,6 +14,7 @@ gym[atari]==0.18.3 h5py==3.1.0; python_version < '3.7' h5py==3.3.0; python_version >= '3.7' hpbandster==0.7.4 +pymoo<0.5.0 # this is a HEBO dependency, remove after https://github.com/huawei-noah/noah-research/issues/41 is fixed HEBO==0.1.0 hyperopt==0.2.5 jupyter==1.0.0 @@ -28,12 +29,11 @@ nevergrad==0.4.3.post7 optuna==2.9.1 pytest-remotedata==0.3.2 lightning-bolts==0.3.3 -pytorch-lightning==1.4.5 +pytorch-lightning==1.4.7 scikit-learn==0.24.2 scikit-optimize==0.8.1 sigopt==7.5.0 smart_open==5.1.0 -tensorflow-probability==0.13.0 timm==0.4.5