Browse Source

Added seed command-line argument

Maxime Chevalier-Boisvert 5 years ago
parent
commit
620b9e1e48
1 changed files with 9 additions and 0 deletions
  1. 9 0
      manual_control.py

+ 9 - 0
manual_control.py

@@ -15,6 +15,9 @@ def redraw(img):
     window.show_img(img)
 
 def reset():
+    if args.seed != -1:
+        env.seed(args.seed)
+
     obs = env.reset()
 
     if hasattr(env, 'mission'):
@@ -76,6 +79,12 @@ parser.add_argument(
     default='MiniGrid-MultiRoom-N6-v0'
 )
 parser.add_argument(
+    "--seed",
+    type=int,
+    help="random seed to generate the environment with",
+    default=-1
+)
+parser.add_argument(
     "--tile_size",
     type=int,
     help="size at which to render tiles",