Browse Source

add info about opened_door

refactoring
sp 10 months ago
parent
commit
59f0011614
  1. 5
      examples/shields/rl/sb3utils.py

5
examples/shields/rl/sb3utils.py

@ -102,6 +102,7 @@ class InfoCallback(BaseCallback):
self.sum_goal = 0 self.sum_goal = 0
self.sum_lava = 0 self.sum_lava = 0
self.sum_collisions = 0 self.sum_collisions = 0
self.sum_opened_door = 0
def _on_step(self) -> bool: def _on_step(self) -> bool:
infos = self.locals["infos"][0] infos = self.locals["infos"][0]
@ -115,4 +116,8 @@ class InfoCallback(BaseCallback):
if infos["collision"]: if infos["collision"]:
self.sum_collisions += 1 self.sum_collisions += 1
self.logger.record("info/sum_collision", self.sum_collisions) self.logger.record("info/sum_collision", self.sum_collisions)
if "opened_door" in infos:
if infos["opened_door"]:
self.sum_opened_door += 1
self.logger.record("info/sum_opened_door", self.sum_opened_door)
return True return True
Loading…
Cancel
Save