Browse Source

pass prism file to handler

refactoring
sp 6 months ago
parent
commit
493f66a950
  1. 2
      examples/shields/rl/13_minigridsb.py

2
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")

Loading…
Cancel
Save