Explorar o código

remove reset seeding

Rodrigo Perez-Vicente %!s(int64=2) %!d(string=hai) anos
pai
achega
09c12b1dab
Modificáronse 1 ficheiros con 3 adicións e 3 borrados
  1. 3 3
      gym_minigrid/manual_control.py

+ 3 - 3
gym_minigrid/manual_control.py

@@ -15,8 +15,7 @@ def redraw(img):
 
 
 def reset():
-    seed = None if args.seed == -1 else args.seed
-    obs = env.reset(seed=seed)
+    obs = env.reset()
 
     if hasattr(env, "mission"):
         print("Mission: %s" % env.mission)
@@ -92,7 +91,8 @@ parser.add_argument(
 
 args = parser.parse_args()
 
-env = gym.make(args.env)
+seed = None if args.seed == -1 else args.seed
+env = gym.make(args.env, seed=seed)
 
 if args.agent_view:
     env = RGBImgPartialObsWrapper(env)