fixes bug in way SDC info is displayed via prettify_table when variab…

…les have spaces
AI-SDC · Mar 22, 2024 · 6f7b561 · 6f7b561
1 parent 444209b
commit 6f7b561
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]