Browse Source

changed default env

refactoring
sp 11 months ago
parent
commit
7eeb816013
  1. 8
      examples/shields/rl/utils.py

8
examples/shields/rl/utils.py

@ -281,17 +281,13 @@ def parse_arguments(argparse):
# parser.add_argument("--env", help="gym environment to load", default="MiniGrid-Empty-8x8-v0") # parser.add_argument("--env", help="gym environment to load", default="MiniGrid-Empty-8x8-v0")
parser.add_argument("--env", parser.add_argument("--env",
help="gym environment to load", help="gym environment to load",
default="MiniGrid-LavaSlipperyS12-v2",
default="MiniGrid-LavaSlipperyCliffS12-v2",
choices=[ choices=[
"MiniGrid-Adv-8x8-v0", "MiniGrid-Adv-8x8-v0",
"MiniGrid-AdvSimple-8x8-v0", "MiniGrid-AdvSimple-8x8-v0",
"MiniGrid-LavaCrossingS9N1-v0", "MiniGrid-LavaCrossingS9N1-v0",
"MiniGrid-LavaCrossingS9N3-v0", "MiniGrid-LavaCrossingS9N3-v0",
"MiniGrid-LavaSlipperyS12-v0",
"MiniGrid-LavaSlipperyS12-v1",
"MiniGrid-LavaSlipperyS12-v2",
"MiniGrid-LavaSlipperyS12-v3",
"MiniGrid-LavaSlipperyCliffS12-v0"
]) ])
# parser.add_argument("--seed", type=int, help="seed for environment", default=None) # parser.add_argument("--seed", type=int, help="seed for environment", default=None)

Loading…
Cancel
Save