From 493f66a95025cbaf8ab27494997870dce0cc27f6 Mon Sep 17 00:00:00 2001 From: sp Date: Mon, 1 Jul 2024 17:48:09 +0200 Subject: [PATCH] pass prism file to handler --- examples/shields/rl/13_minigridsb.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/shields/rl/13_minigridsb.py b/examples/shields/rl/13_minigridsb.py index 1cd0a7e..2170a9e 100644 --- a/examples/shields/rl/13_minigridsb.py +++ b/examples/shields/rl/13_minigridsb.py @@ -36,7 +36,7 @@ def main(): if shield_needed(args.shielding): - shield_handler = MiniGridShieldHandler(GRID_TO_PRISM_BINARY, args.grid_file, args.prism_output_file, formula, shield_value=shield_value, shield_comparison=shield_comparison, nocleanup=args.nocleanup, prism_file=None) + shield_handler = MiniGridShieldHandler(GRID_TO_PRISM_BINARY, args.grid_file, args.prism_output_file, formula, shield_value=shield_value, shield_comparison=shield_comparison, nocleanup=args.nocleanup, prism_file=args.prism_file) env = gym.make(args.env, render_mode="rgb_array")