fix: Restore proper string serialization of DocItemLabel (#124)

* feat: Add mapping to colors into DocItemLabel Signed-off-by: Christoph Auer <[email protected]> * fix: Restore DocItemLabel.__str__ Signed-off-by: Christoph Auer <[email protected]> --------- Signed-off-by: Christoph Auer <[email protected]>
DS4SD · Jan 8, 2025 · a52bb88 · a52bb88
1 parent 5c2e0de
commit a52bb88
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docling_core/types/doc/labels.py b/docling_core/types/doc/labels.py
@@ -29,6 +29,10 @@ class DocItemLabel(str, Enum):
     PARAGRAPH = "paragraph"
     REFERENCE = "reference"
 
+    def __str__(self):
+        """Get string value."""
+        return str(self.value)
+
     @staticmethod
     def get_color(label: "DocItemLabel") -> Tuple[int, int, int]:
         """Return the RGB color associated with a given label."""