Browse Source

add seed to reset manual_control

Rodrigo Perez-Vicente 2 năm trước cách đây
mục cha
commit
0e4462c8a0
1 tập tin đã thay đổi với 5 bổ sung6 xóa
  1. 5 6
      gym_minigrid/manual_control.py

+ 5 - 6
gym_minigrid/manual_control.py

@@ -10,8 +10,8 @@ def redraw(window, img):
     window.show_img(img)
     window.show_img(img)
 
 
 
 
-def reset(env, window):
-    _ = env.reset()
+def reset(env, window, seed=None):
+    _ = env.reset(seed=seed)
 
 
     if hasattr(env, "mission"):
     if hasattr(env, "mission"):
         print("Mission: %s" % env.mission)
         print("Mission: %s" % env.mission)
@@ -99,10 +99,8 @@ if __name__ == "__main__":
 
 
     args = parser.parse_args()
     args = parser.parse_args()
 
 
-    seed = None if args.seed == -1 else args.seed
     env = gym.make(
     env = gym.make(
         args.env,
         args.env,
-        seed=seed,
         new_step_api=True,
         new_step_api=True,
         render_mode="human",  # Note that we do not need to use "human", as Window handles human rendering.
         render_mode="human",  # Note that we do not need to use "human", as Window handles human rendering.
         tile_size=args.tile_size,
         tile_size=args.tile_size,
@@ -114,8 +112,9 @@ if __name__ == "__main__":
 
 
     window = Window("gym_minigrid - " + args.env)
     window = Window("gym_minigrid - " + args.env)
     window.reg_key_handler(lambda event: key_handler(env, window, event))
     window.reg_key_handler(lambda event: key_handler(env, window, event))
-
-    reset(env, window)
+    
+    seed = None if args.seed == -1 else args.seed
+    reset(env, window, seed)
 
 
     # Blocking event loop
     # Blocking event loop
     window.show(block=True)
     window.show(block=True)