Make ingest work with label annotation type objects (#753)

kolenaIO · Feb 6, 2025 · ed91015 · ed91015
1 parent a300c1c
commit ed91015
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/kolena/_experimental/classification/_matching.py b/kolena/_experimental/classification/_matching.py
@@ -21,8 +21,7 @@
 from typing import TypeVar
 from typing import Union
 
-from pydantic.dataclasses import dataclass
-
+from kolena._utils.pydantic_v1.dataclasses import dataclass
 from kolena.annotation import Label
 from kolena.annotation import ScoredLabel
 

diff --git a/kolena/_utils/datatypes.py b/kolena/_utils/datatypes.py
@@ -292,6 +292,8 @@ def toDict(self) -> Dict:
 
 
 def _serialize_dataobject(x: Any) -> Any:
+    if isinstance(x, dict):
+        return {k: _serialize_dataobject(v) for k, v in x.items()}
     if isinstance(x, list):
         return [item._to_dict() if isinstance(item, DataObject) else item for item in x]