Browse Source

disabled shield creation at reset WIP

refactoring
sp 10 months ago
parent
commit
4486dba2c9
  1. 2
      examples/shields/rl/rllibutils.py

2
examples/shields/rl/rllibutils.py

@ -97,7 +97,7 @@ class MiniGridShieldingWrapper(gym.core.Wrapper):
} }
) )
self.shield_creator = shield_creator 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.shield = shield_creator.create_shield(env=self.env)
self.mask_actions = mask_actions self.mask_actions = mask_actions
self.shield_query_creator = shield_query_creator self.shield_query_creator = shield_query_creator

Loading…
Cancel
Save