fix: str representation of enum across python versions (#60)

* fix: enforce consistent output across linux, macos and windows Signed-off-by: Peter Staar <[email protected]> * reformat docling-core Signed-off-by: Peter Staar <[email protected]> * added __str__ to enum class and removed the .name from function Signed-off-by: Peter Staar <[email protected]> --------- Signed-off-by: Peter Staar <[email protected]>
DS4SD · Oct 29, 2024 · 8528918 · 8528918
1 parent 4993c34
commit 8528918
Showing 1 changed file with 12 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"  # explicitly a paragraph and not arbitrary text
     REFERENCE = "reference"
 
+    def __str__(self):
+        """Get string value."""
+        return str(self.value)
+
 
 class GroupLabel(str, Enum):
     """GroupLabel."""
@@ -43,6 +47,10 @@ class GroupLabel(str, Enum):
     SHEET = "sheet"
     SLIDE = "slide"
 
+    def __str__(self):
+        """Get string value."""
+        return str(self.value)
+
 
 class TableCellLabel(str, Enum):
     """TableCellLabel."""
@@ -51,3 +59,7 @@ class TableCellLabel(str, Enum):
     ROW_HEADER = "row_header"
     ROW_SECTION = "row_section"
     BODY = "body"
+
+    def __str__(self):
+        """Get string value."""
+        return str(self.value)