From b9adb88654a14049a423f6f0d3187c6748c1c339 Mon Sep 17 00:00:00 2001 From: Omar Younis <42100908+younik@users.noreply.github.com> Date: Fri, 7 Oct 2022 16:28:45 +0200 Subject: [PATCH 1/2] update rl tutorial to last gym version --- intermediate_source/reinforcement_q_learning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intermediate_source/reinforcement_q_learning.py b/intermediate_source/reinforcement_q_learning.py index 6a03d61f79e..e5d9765b8e6 100644 --- a/intermediate_source/reinforcement_q_learning.py +++ b/intermediate_source/reinforcement_q_learning.py @@ -79,7 +79,7 @@ import torchvision.transforms as T -env = gym.make('CartPole-v0', new_step_api=True, render_mode='single_rgb_array').unwrapped +env = gym.make('CartPole-v0', render_mode='rgb_array').unwrapped # set up matplotlib is_ipython = 'inline' in matplotlib.get_backend() From 4c72b1bbc1418d48dc2d88632e975bc739ae4d48 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Sun, 9 Oct 2022 09:38:51 -0700 Subject: [PATCH 2/2] Make it compatible with 0.26 and 0.25 gym versions --- intermediate_source/reinforcement_q_learning.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/intermediate_source/reinforcement_q_learning.py b/intermediate_source/reinforcement_q_learning.py index e5d9765b8e6..8ba967f6575 100644 --- a/intermediate_source/reinforcement_q_learning.py +++ b/intermediate_source/reinforcement_q_learning.py @@ -79,7 +79,10 @@ import torchvision.transforms as T -env = gym.make('CartPole-v0', render_mode='rgb_array').unwrapped +if gym.__version__ < '0.26': + env = gym.make('CartPole-v0', new_step_api=True, render_mode='single_rgb_array').unwrapped +else: + env = gym.make('CartPole-v0', render_mode='rgb_array').unwrapped # set up matplotlib is_ipython = 'inline' in matplotlib.get_backend()