From c8e76f9b924fc61123bb404ad8c6c4bd023a9d10 Mon Sep 17 00:00:00 2001
From: Amog Kamsetty <amogkam@users.noreply.github.com>
Date: Fri, 18 Feb 2022 17:39:22 -0800
Subject: [PATCH] [rllib] Upper bound `gym` version (#22510)

gym had 0.22 release today which is breaking a lot of the rllib tests and examples. Temporarily pins gym version for now.
---
 python/requirements.txt                       | 2 +-
 python/requirements/ml/requirements_rllib.txt | 2 +-
 python/setup.py                               | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/requirements.txt b/python/requirements.txt
index a76676f24..16f3608f7 100644
--- a/python/requirements.txt
+++ b/python/requirements.txt
@@ -30,7 +30,7 @@ smart_open
 ## setup.py extras
 dm_tree
 flask
-gym>=0.21.0; python_version >= '3.7'
+gym>=0.21.0,<0.22.0; python_version >= '3.7'
 gym==0.19.0; python_version < '3.7'
 lz4
 scikit-image
diff --git a/python/requirements/ml/requirements_rllib.txt b/python/requirements/ml/requirements_rllib.txt
index 3c182da51..5da913cf9 100644
--- a/python/requirements/ml/requirements_rllib.txt
+++ b/python/requirements/ml/requirements_rllib.txt
@@ -4,7 +4,7 @@
 # ---------------------
 # Atari
 autorom[accept-rom-license]
-gym[atari]>=0.21.0; python_version >= '3.7'
+gym[atari]>=0.21.0,<0.22.0; python_version >= '3.7'
 gym[atari]==0.19.0; python_version < '3.7'
 # Kaggle envs.
 kaggle_environments==1.7.11
diff --git a/python/setup.py b/python/setup.py
index e26d55f42..dbf0b3b10 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -240,7 +240,7 @@ if setup_spec.type == SetupType.RAY:
 
     setup_spec.extras["rllib"] = setup_spec.extras["tune"] + [
         "dm_tree",
-        "gym",
+        "gym<0.22",
         "lz4",
         # matplotlib (dependency of scikit-image) 3.4.3 breaks docker build
         # Todo: Remove this when safe?