Add colon at the end of single line headers

emdgroup · Sep 6, 2024 · 646fb95 · 646fb95
1 parent 6e1cac9
commit 646fb95
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/baybe/utils/plotting.py b/baybe/utils/plotting.py
@@ -169,6 +169,8 @@ def to_string(header: str, *fields: Any, single_line: bool = False) -> str:
             raise ValueError(
                 "``single_line`` is only applicable when given a single field."
             )
+        # Since single line headers look ugly without a ":", we add it manually
+        header = header if header.endswith(":") else header + ":"
         return f"{header} {str(fields[0])}"
 
     return "\n".join([header] + [indent(str(f)) for f in fields])