diff --git a/vmas/scenarios/joint_passage.py b/vmas/scenarios/joint_passage.py index cea4268a..f5b7b27d 100644 --- a/vmas/scenarios/joint_passage.py +++ b/vmas/scenarios/joint_passage.py @@ -458,11 +458,11 @@ def reward(self, agent: Agent): self.world.get_distance(a, passage) <= self.min_collision_distance ] += self.collision_reward - for wall in self.walls: - self.collision_rew[ - self.world.get_distance(a, wall) - <= self.min_collision_distance - ] += self.collision_reward + for wall in self.walls: + self.collision_rew[ + self.world.get_distance(a, wall) + <= self.min_collision_distance + ] += self.collision_reward # Joint collisions for p in self.passages: