Skip to content

Commit b9adb88

Browse files
authored
update rl tutorial to last gym version
1 parent 1a946b0 commit b9adb88

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/reinforcement_q_learning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@
7979
import torchvision.transforms as T
8080

8181

82-
env = gym.make('CartPole-v0', new_step_api=True, render_mode='single_rgb_array').unwrapped
82+
env = gym.make('CartPole-v0', render_mode='rgb_array').unwrapped
8383

8484
# set up matplotlib
8585
is_ipython = 'inline' in matplotlib.get_backend()

0 commit comments

Comments
 (0)