From af0c4e2f210ef5c282d326ce429345bb2b2011b8 Mon Sep 17 00:00:00 2001 From: Thomas Knoll Date: Mon, 18 Dec 2023 22:40:15 +0100 Subject: [PATCH] shield value script 15 --- examples/shields/rl/15_train_eval_tune.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/shields/rl/15_train_eval_tune.py b/examples/shields/rl/15_train_eval_tune.py index 6a8f0df..0158e08 100644 --- a/examples/shields/rl/15_train_eval_tune.py +++ b/examples/shields/rl/15_train_eval_tune.py @@ -31,7 +31,9 @@ def shielding_env_creater(config): shield_creator = MiniGridShieldHandler(grid_file=args.grid_path, grid_to_prism_path=args.grid_to_prism_binary_path, prism_path=args.prism_path, - formula=args.formula) + formula=args.formula, + shield_value=args.shield_value, + prism_config=args.prism_config) env = gym.make(name, randomize_start=True) env = MiniGridShieldingWrapper(env, shield_creator=shield_creator, shield_query_creator=create_shield_query ,mask_actions=shielding != ShieldingConfig.Disabled)