update rl tutorial to last gym version (#2073)

younik · malfet · web-flow · commit 2047596b3c31 · 2022-10-09T11:25:19.000-07:00
* update rl tutorial to last gym version
* Make it compatible with 0.26 and 0.25 gym versions

Co-authored-by: Nikita Shulga &lt;nikita.shulga@gmail.com&gt;
diff --git 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', new_step_api=True, render_mode='single_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()
