fixed bug in info callback

11 months ago · 62fddc1e27
1 changed files with 1 additions and 1 deletions
--- 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