|
@ -10,7 +10,7 @@ class MiniGridSbShieldingWrapper(gym.core.Wrapper): |
|
|
def __init__(self, |
|
|
def __init__(self, |
|
|
env, |
|
|
env, |
|
|
shield_handler : MiniGridShieldHandler, |
|
|
shield_handler : MiniGridShieldHandler, |
|
|
create_shield_at_reset = True, |
|
|
|
|
|
|
|
|
create_shield_at_reset = False, |
|
|
): |
|
|
): |
|
|
super().__init__(env) |
|
|
super().__init__(env) |
|
|
self.shield_handler = shield_handler |
|
|
self.shield_handler = shield_handler |
|
|