Merge pull request #202 from AI-SDC/200-prettify_tablestringdata-mess…

…es-column-parsing-if-first-variable-value-has-spaces fixes bug in way SDC info is displayed via prettify_table
AI-SDC · Mar 28, 2024 · 2d09424 · 2d09424
2 parents 444209b + 6f7b561
commit 2d09424
Show file tree

Hide file tree

Showing 3 changed files with 197 additions and 3,105 deletions.
diff --git a/acro/utils.py b/acro/utils.py
@@ -46,6 +46,7 @@ def prettify_table_string(table: pd.DataFrame, separator: str | None = None) ->
     hdelim = "-"
     vdelim = "|"
 
+    table.rename(columns=lambda x: str(x).replace(" ", "_"), inplace=True)
     output = table.to_string(justify="left")
     as_strings = output.split("\n")
     nheaders = len(as_strings) - table.shape[0]