feat: implement additional parameters in Metadata

raillabel/format/metadata.py

@@ -39,7 +39,7 @@ class Metadata:
     @classmethod
     def from_json(cls, json: JSONMetadata) -> Metadata:
         """Construct an instant of this class from RailLabel JSON data."""
-        return Metadata(
+        metadata = Metadata(
             schema_version=json.schema_version,
             name=json.name,
             subschema_version=json.subschema_version,
@@ -49,3 +49,9 @@ def from_json(cls, json: JSONMetadata) -> Metadata:
             annotator=json.annotator,
             comment=json.comment,
         )
+
+        if json.model_extra is not None:
+            for extra_field, extra_value in json.model_extra.items():
+                setattr(metadata, extra_field, extra_value)
+
+        return metadata

raillabel/json_format/metadata.py

@@ -5,10 +5,10 @@
 
 from typing import Literal
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Extra
 
 
-class JSONMetadata(BaseModel):
+class JSONMetadata(BaseModel, extra=Extra.allow):
     """Metadata about the annotation file itself."""
 
     schema_version: Literal["1.0.0"]

tests/test_raillabel/format/test_metadata.py

@@ -47,5 +47,19 @@ def test_from_json(metadata, metadata_json):
     assert actual == metadata
 
 
+def test_from_json__extra_fields():
+    json_metadata = JSONMetadata(
+        **{
+            "schema_version": "1.0.0",
+            "ADDITIONAL_STR": "SOME_VALUE",
+            "ADDITIONAL_OBJECT": {"first_field": 2, "second_field": [1, 2, 3]},
+        }
+    )
+
+    actual = Metadata.from_json(json_metadata)
+    assert actual.ADDITIONAL_STR == "SOME_VALUE"
+    assert actual.ADDITIONAL_OBJECT == {"first_field": 2, "second_field": [1, 2, 3]}
+
+
 if __name__ == "__main__":
     pytest.main([__file__, "-v"])