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