fix: Refactor validate_rail_side for raillabel 4

raillabel_providerkit/validation/validate_rail_side/validate_rail_side.py

@@ -3,6 +3,8 @@
 
 from __future__ import annotations
 
+from uuid import UUID
+
 import numpy as np
 import raillabel
 from raillabel.filter import (
@@ -31,7 +33,7 @@ def validate_rail_side(scene: raillabel.Scene) -> list[str]:
     errors: list[str] = []
 
     # Get a list of camera uids
-    cameras = list(scene.filter([IncludeSensorTypeFilter("camera")]).sensors.keys())
+    cameras = list(scene.filter([IncludeSensorTypeFilter(["camera"])]).sensors.keys())
 
     # Check per camera
     for camera in cameras:
@@ -123,9 +125,9 @@ def _check_rails_for_swap(
     return None
 
 
-def _count_rails_per_track_in_frame(frame: raillabel.format.Frame) -> dict[str, tuple[int, int]]:
+def _count_rails_per_track_in_frame(frame: raillabel.format.Frame) -> dict[UUID, tuple[int, int]]:
     # For each track, the left and right rail counts are stored as a list (left, right)
-    counts: dict[str, list[int, int]] = {}
+    counts: dict[UUID, list[int]] = {}
 
     # For each track, count the left and right rails
     unfiltered_annotations = list(frame.annotations.values())
@@ -148,11 +150,14 @@ def _count_rails_per_track_in_frame(frame: raillabel.format.Frame) -> dict[str,
             continue
 
     # Return results
-    return {key: tuple(value) for key, value in counts.items()}
+    return {
+        object_id: (object_counts[0], object_counts[1])
+        for object_id, object_counts in counts.items()
+    }
 
 
 def _filter_for_poly2ds(
-    unfiltered_annotations: list[type[raillabel.format._ObjectAnnotation]],
+    unfiltered_annotations: list,
 ) -> list[raillabel.format.Poly2d]:
     return [
         annotation
@@ -271,9 +276,12 @@ def _find_x_by_y(y: float, poly2d: raillabel.format.Poly2d) -> float | None:
 
 
 def _get_track_from_frame(
-    frame: raillabel.format.Frame, object_uid: str, rail_side: str
+    frame: raillabel.format.Frame, object_uid: UUID, rail_side: str
 ) -> raillabel.format.Poly2d | None:
     for annotation in frame.annotations.values():
+        if not isinstance(annotation, raillabel.format.Poly2d):
+            continue
+
         if annotation.object_id != object_uid:
             continue