Added pygame now required by gym and cartpole env (#2485)

pytorch · Feb 19, 2022 · e2c7ec8 · e2c7ec8
1 parent 6d385eb
commit e2c7ec8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/examples/reinforcement_learning/actor_critic.py b/examples/reinforcement_learning/actor_critic.py
@@ -146,7 +146,7 @@ def should_finish_training(engine):
     )
     args = parser.parse_args()
 
-    env = gym.make("CartPole-v0")
+    env = gym.make("CartPole-v1")
     env.seed(args.seed)
     torch.manual_seed(args.seed)
 

diff --git a/examples/reinforcement_learning/reinforce.py b/examples/reinforcement_learning/reinforce.py
@@ -136,7 +136,7 @@ def should_finish_training(engine):
     )
     args = parser.parse_args()
 
-    env = gym.make("CartPole-v0")
+    env = gym.make("CartPole-v1")
     env.seed(args.seed)
     torch.manual_seed(args.seed)
 

diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -31,4 +31,6 @@ nltk
 # Examples dependencies
 pandas
 gym
+# pygame is now required by gym for cartpole env
+pygame
 mkl
-Original file line number
+Diff line change
@@ Expand Up / @@ -31,4 +31,6 @@ nltk @@
     # Examples dependencies
     pandas
     gym
+    # pygame is now required by gym for cartpole env
+    pygame
     mkl