Fixed Error in update_road_ids

Author: sventarika

omega_prime/mapsemgents.py

@@ -80,10 +80,23 @@ def update_road_ids(self):
         """
         Updates the road_ids of the lane to the segment ID
         """
-        for i, lane in enumerate(self.lanes.values()):
+        # Create a list of items to avoid modifying dictionary during iteration
+        lanes_items = list(self.lanes.items())
+        
+        for original_key, lane in lanes_items:
             lane_id = lane.idx.lane_id
             if lane_id in self.lane_segment_dict and self.lane_segment_dict[lane_id].segment is not None:
-                lane.idx = OsiLaneId(road_id=self.lane_segment_dict[lane_id].segment.idx, lane_id=lane_id)
+                # Create the new namedtuple with updated road_id
+                new_idx = lane.idx._replace(road_id=self.lane_segment_dict[lane_id].segment.idx)
+                
+                # Create a new lane object with the updated idx
+                updated_lane = lane
+                updated_lane.idx = new_idx
+                
+                # Update the dictionary - remove old entry and add new one
+                self.lanes[new_idx] = updated_lane
+                if original_key != new_idx:
+                    del self.lanes[original_key]  # Delete the old key if it's different
             else:
                 # print(f"Lane {lane_id} is not part of an intersection, so it has no segment ID")
                 pass