fix failing test

luxonis · Nov 5, 2024 · 4f33d4f · 4f33d4f
1 parent 4832697
commit 4f33d4f
Showing 1 changed file with 28 additions and 26 deletions.
diff --git a/tests/test_data/test_dataset.py b/tests/test_data/test_dataset.py
@@ -481,37 +481,39 @@ def generator():
         "vehicle_segmentation",
     }
     assert dataset.get_classes()[1] == {
-        "boundingbox": ["motorbike", "car"],
+        "boundingbox": sorted(["motorbike", "car"]),
         "motorbike-boundingbox": ["motorbike"],
         "motorbike-keypoints": ["motorbike"],
         "car-boundingbox": ["car"],
         "car-keypoints": ["car"],
         "vehicle_type_segmentation": ["background", "car", "motorbike"],
-        "brand_segmentation": [
-            "background",
-            "chrysler",
-            "bmw",
-            "ducati",
-            "dodge",
-            "ferrari",
-            "infiniti",
-            "land-rover",
-            "roll-royce",
-            "saab",
-            "Kawasaki",
-            "moto",
-            "truimph",
-            "alfa-romeo",
-            "harley",
-            "honda",
-            "jeep",
-            "aprilia",
-            "piaggio",
-            "yamaha",
-            "buick",
-            "pontiac",
-            "isuzu",
-        ],
+        "brand_segmentation": sorted(
+            [
+                "background",
+                "chrysler",
+                "bmw",
+                "ducati",
+                "dodge",
+                "ferrari",
+                "infiniti",
+                "land-rover",
+                "roll-royce",
+                "saab",
+                "Kawasaki",
+                "moto",
+                "truimph",
+                "alfa-romeo",
+                "harley",
+                "honda",
+                "jeep",
+                "aprilia",
+                "piaggio",
+                "yamaha",
+                "buick",
+                "pontiac",
+                "isuzu",
+            ]
+        ),
         "color_segmentation": ["background", "blue", "green", "red"],
         "vehicle_segmentation": ["vehicle"],
     }