From 7eeb816013f674fe8839bfae06ec7f007d85c340 Mon Sep 17 00:00:00 2001 From: sp Date: Wed, 3 Jan 2024 19:51:16 +0100 Subject: [PATCH] changed default env --- examples/shields/rl/utils.py | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) diff --git a/examples/shields/rl/utils.py b/examples/shields/rl/utils.py index 0816a54..5bb3fcf 100644 --- a/examples/shields/rl/utils.py +++ b/examples/shields/rl/utils.py @@ -265,35 +265,31 @@ def get_action_index_mapping(actions): elif "pickup" in action_str: return Actions.pickup elif "done" in action_str: - return Actions.done + return Actions.done elif "drop" in action_str: return Actions.drop elif "toggle" in action_str: return Actions.toggle elif "unlock" in action_str: return Actions.toggle - + raise ValueError("No action mapping found") - + def parse_arguments(argparse): parser = argparse.ArgumentParser() # 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-LavaSlipperyS12-v2", + parser.add_argument("--env", + help="gym environment to load", + default="MiniGrid-LavaSlipperyCliffS12-v2", choices=[ "MiniGrid-Adv-8x8-v0", "MiniGrid-AdvSimple-8x8-v0", "MiniGrid-LavaCrossingS9N1-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("--grid_to_prism_binary_path", default="./main") parser.add_argument("--grid_path", default="grid")