[BugFix] Occupied positions flocking

matteobettini · matteobettini · commit 3e82ec7b8b1e · 2023-12-04T16:01:47.000Z
diff --git a/vmas/scenarios/flocking.py b/vmas/scenarios/flocking.py
@@ -101,6 +101,7 @@ def reset_world_at(self, env_index: int = None):
             self._min_dist_between_entities,
             x_bounds=(-self.x_dim, self.x_dim),
             y_bounds=(-self.y_dim, self.y_dim),
+            occupied_positions=target_pos.unsqueeze(1),
         )
 
         for agent in self.world.policy_agents:

Original file line number	Diff line number	Diff line change
`@@ -101,6 +101,7 @@ def reset_world_at(self, env_index: int = None):`
`101`	`101`	`self._min_dist_between_entities,`
`102`	`102`	`x_bounds=(-self.x_dim, self.x_dim),`
`103`	`103`	`y_bounds=(-self.y_dim, self.y_dim),`
	`104`	`+ occupied_positions=target_pos.unsqueeze(1),`
`104`	`105`	`)`
`105`	`106`
`106`	`107`	`for agent in self.world.policy_agents:`