diff --git a/examples/shields/rl/sb3utils.py b/examples/shields/rl/sb3utils.py index 9092997..dcf20d2 100644 --- a/examples/shields/rl/sb3utils.py +++ b/examples/shields/rl/sb3utils.py @@ -113,6 +113,6 @@ class InfoCallback(BaseCallback): self.logger.record("info/sum_ran_into_lava", self.sum_lava) if "collision" in infos: if infos["collision"]: - self.sum_collision += 1 + self.sum_collisions += 1 self.logger.record("info/sum_collision", self.sum_collisions) return True