From d71d752a846c0f382deb3de75efeef3c592b5435 Mon Sep 17 00:00:00 2001 From: KieranRatcliffeInvertedAI <kieran.ratcliffe@inverted.ai> Date: Thu, 20 Jun 2024 15:58:04 -0700 Subject: [PATCH] Fix to large_initialize display option. --- invertedai/large/initialize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/invertedai/large/initialize.py b/invertedai/large/initialize.py index d9a5ae69..2a9b85d1 100644 --- a/invertedai/large/initialize.py +++ b/invertedai/large/initialize.py @@ -195,7 +195,7 @@ def get_number_of_agents_per_region_by_drivable_area( desc=f"Calculating drivable surface areas" ) else: - iterable_regions = new_regions + iterable_regions = enumerate(new_regions) for i, region in iterable_regions: center_tuple = (region.center.x, region.center.y) @@ -312,7 +312,7 @@ def inside_fov(center: Point, agent_scope_fov: float, point: Point) -> bool: desc=f"Initializing regions" ) else: - iterable_regions = regions + iterable_regions = enumerate(regions) num_attempts = 1 + len(regions) // ATTEMPT_PER_NUM_REGIONS all_responses = []