From 62fddc1e2712bbfd32c5ce8ebf3cfedf1238ebe8 Mon Sep 17 00:00:00 2001 From: sp Date: Thu, 18 Jan 2024 22:33:40 +0100 Subject: [PATCH] fixed bug in info callback --- examples/shields/rl/sb3utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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