From 4486dba2c9b34488dd224cd7e397d7b5d7275512 Mon Sep 17 00:00:00 2001 From: sp Date: Wed, 3 Jan 2024 20:20:31 +0100 Subject: [PATCH] disabled shield creation at reset WIP --- examples/shields/rl/rllibutils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/shields/rl/rllibutils.py b/examples/shields/rl/rllibutils.py index deaf959..a22dda1 100644 --- a/examples/shields/rl/rllibutils.py +++ b/examples/shields/rl/rllibutils.py @@ -97,7 +97,7 @@ class MiniGridShieldingWrapper(gym.core.Wrapper): } ) self.shield_creator = shield_creator - self.create_shield_at_reset = create_shield_at_reset + self.create_shield_at_reset = False # TODO self.shield = shield_creator.create_shield(env=self.env) self.mask_actions = mask_actions self.shield_query_creator = shield_query_creator