Merge pull request #25 from DSD-DBS/uai_loading_fix

Uai loading fix

CHANGELOG.md

@@ -53,3 +53,7 @@ Release
 - Raillabel schema is now more restrictive regarding intrinsic calibration
 - Support for understand.ai t4 format
 - Renaming of raillabel.format.Frame.data to frame_data
+
+### 2.2.1
+- Fixed bug, that prevented loading a understand ai file via raillabel.load()
+- Made the project_id field in the understand ai files less restrictive

raillabel/format_loaders/loader_raillabel_v2.py

@@ -117,6 +117,9 @@ def supports(self, data: dict) -> bool:
             If True, the Loader class is suitable for the data.
         """
 
+        if "openlabel" not in data or "metadata" not in data["openlabel"]:
+            return False
+
         if "subschema_version" in data["openlabel"]["metadata"]:
             return (
                 "openlabel" in data

raillabel/format_loaders/loader_understand_ai_t4.py

@@ -43,6 +43,8 @@ def load(self, data: dict, validate: bool = True) -> uai_format.Scene:
             The loaded scene with the data.
         """
 
+        self.warnings = []
+
         if validate:
             self.validate(data)
 
@@ -69,7 +71,6 @@ def supports(self, data: dict) -> bool:
         return (
             "metadata" in data
             and "project_id" in data["metadata"]
-            and data["metadata"]["project_id"] == "trains_4"
             and "coordinateSystems" in data
             and "frames" in data
         )

tests/test_raillabel/format_loaders/test_loader_understand_ai_t4.py

@@ -23,7 +23,7 @@ def test_supports_true(json_data, loader):
 
 def test_supports_false(json_data, loader):
     data = json_data["understand_ai_real_life"]
-    data["metadata"]["project_id"] = "invalid"
+    del data["metadata"]["project_id"]
     assert not loader.supports(data)
 
 

tests/test_raillabel/test_load.py

@@ -0,0 +1,30 @@
+# Copyright DB Netz AG and contributors
+# SPDX-License-Identifier: Apache-2.0
+
+import os
+import sys
+from pathlib import Path
+
+import pytest
+
+sys.path.insert(1, str(Path(__file__).parent.parent.parent))
+
+import raillabel
+
+
+def test_load_raillabel(json_paths):
+    data_path = json_paths["openlabel_v1_short"]
+    scene = raillabel.load(data_path)
+    assert len(scene.frames) != 0
+
+
+def test_load_uai(json_paths):
+    data_path = json_paths["understand_ai_t4_short"]
+    scene = raillabel.load(data_path)
+    assert len(scene.frames) != 0
+
+
+# Executes the test if the file is called
+if __name__ == "__main__":
+    os.system("clear")
+    pytest.main([__file__, "--disable-pytest-warnings", "--cache-clear"])