From aa2cec0e0f7dbc07a876027959872aad2d15ba22 Mon Sep 17 00:00:00 2001 From: sp Date: Wed, 3 Jan 2024 19:51:42 +0100 Subject: [PATCH] disabled default shield creation on reset --- examples/shields/rl/rllibutils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/shields/rl/rllibutils.py b/examples/shields/rl/rllibutils.py index ea48af3..deaf959 100644 --- a/examples/shields/rl/rllibutils.py +++ b/examples/shields/rl/rllibutils.py @@ -82,11 +82,11 @@ class OneHotShieldingWrapper(gym.core.ObservationWrapper): class MiniGridShieldingWrapper(gym.core.Wrapper): - def __init__(self, - env, + def __init__(self, + env, shield_creator : MiniGridShieldHandler, shield_query_creator, - create_shield_at_reset=True, + create_shield_at_reset=False, mask_actions=True): super(MiniGridShieldingWrapper, self).__init__(env) self.max_available_actions = env.action_space.n