Fix: correct joint-passage reward function (issue #145)

Author: Benabdellah22

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: