diff --git a/python/requirements/tune/requirements_tune.txt b/python/requirements/tune/requirements_tune.txt
index c70765c8af94..5d9083ae2e39 100644
--- a/python/requirements/tune/requirements_tune.txt
+++ b/python/requirements/tune/requirements_tune.txt
@@ -10,7 +10,7 @@ flaml==0.5.2
 freezegun==1.1.0
 gluoncv==0.10.1.post0
 gpy==1.10.0
-gym[atari]==0.18.3
+gym[atari]==0.20.0
 h5py==3.1.0; python_version < '3.7'
 h5py==3.3.0; python_version >= '3.7'
 hpbandster==0.7.4
@@ -33,7 +33,6 @@ 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