Browse Source

resynthesize on reset is default False

refactoring
sp 10 months ago
parent
commit
29799bc52f
  1. 2
      examples/shields/rl/sb3utils.py

2
examples/shields/rl/sb3utils.py

@ -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

Loading…
Cancel
Save