Add rich table width if jupyter in modules

This commit is contained in:
froggleston
2024-08-04 21:43:00 +01:00
parent ce8d03ddce
commit cb4747aed2

View File

@@ -37,8 +37,12 @@ def print_rich_table(
row_to_add: List[Union[str, Text]] = [r if isinstance(r, Text) else str(r) for r in row]
table.add_row(*row_to_add)
width = None
if any(module in ["pytest", "ipykernel"] for module in sys.modules):
width = 200
console = Console(
width=200 if "pytest" in sys.modules else None,
width=width
)
console.print(table)
@@ -71,7 +75,11 @@ def print_df_rich_table(
row = [_format_value(x, floatfmt=".3f") for x in value_list]
table.add_row(*row)
width = None
if any(module in ["pytest", "ipykernel"] for module in sys.modules):
width = 200
console = Console(
width=200 if "pytest" in sys.modules else None,
width=width
)
console.print(table)