Skip to content

Commit bbf283e

Browse files
committed
Merge branch 'main' into copyright
# Conflicts: # vmas/scenarios/joint_passage_size.py
2 parents 77aed0e + 6761c7a commit bbf283e

File tree

2 files changed

+8
-10
lines changed

2 files changed

+8
-10
lines changed

vmas/scenarios/joint_passage.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -459,11 +459,10 @@ def reward(self, agent: Agent):
459459
self.world.get_distance(a, passage)
460460
<= self.min_collision_distance
461461
] += self.collision_reward
462-
for wall in self.walls:
463-
self.collision_rew[
464-
self.world.get_distance(a, wall)
465-
<= self.min_collision_distance
466-
] += self.collision_reward
462+
for wall in self.walls:
463+
self.collision_rew[
464+
self.world.get_distance(a, wall) <= self.min_collision_distance
465+
] += self.collision_reward
467466

468467
# Joint collisions
469468
for p in self.passages:

vmas/scenarios/joint_passage_size.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -454,11 +454,10 @@ def reward(self, agent: Agent):
454454
self.world.get_distance(a, passage)
455455
<= self.min_collision_distance
456456
] += self.collision_reward
457-
for wall in self.walls:
458-
self.collision_rew[
459-
self.world.get_distance(a, wall)
460-
<= self.min_collision_distance
461-
] += self.collision_reward
457+
for wall in self.walls:
458+
self.collision_rew[
459+
self.world.get_distance(a, wall) <= self.min_collision_distance
460+
] += self.collision_reward
462461

463462
# Energy reward
464463
if self.energy_reward_coeff != 0:

0 commit comments

Comments
 (0)