disabled default shield creation on reset

2 years ago · aa2cec0e0f
1 changed files with 3 additions and 3 deletions
--- a/examples/shields/rl/rllibutils.py
+++ b/examples/shields/rl/rllibutils.py
@ -86,7 +86,7 @@ class MiniGridShieldingWrapper(gym.core.Wrapper):
                 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