fix: disable number parsing in multi-type columns (#157)

Try formatting a table with number parsing and formatting and, if mixed types, retry with disabled number parsing. Signed-off-by: Cesar Berrospi Ramis <75900930+ceberam@users.noreply.github.com>
DS4SD · Jan 16, 2024 · 667e7fc · 667e7fc
1 parent 192ef9c
commit 667e7fc
Show file tree

Hide file tree

Showing 3 changed files with 67,569 additions and 4 deletions.
diff --git a/deepsearch/documents/core/export.py b/deepsearch/documents/core/export.py
@@ -59,10 +59,17 @@ def export_to_markdown(document: Dict[str, Any]) -> str:
                 table.append(tmp)
 
             if len(table) > 1 and len(table[0]) > 0:
-
-                markdown_text += tabulate(
-                    table[1:], headers=table[0], tablefmt="github"
-                )
+                try:
+                    md_table = tabulate(table[1:], headers=table[0], tablefmt="github")
+                except ValueError:
+                    md_table = tabulate(
+                        table[1:],
+                        headers=table[0],
+                        tablefmt="github",
+                        disable_numparse=True,
+                    )
+
+                markdown_text += md_table
                 markdown_text += "\n\n"
 
     return markdown_text