From 8e780d3a6a0411e3f5af72c343fb542dcf597e9d Mon Sep 17 00:00:00 2001 From: Manfei <41607353+ManfeiBai@users.noreply.github.com> Date: Wed, 22 Nov 2023 15:17:34 -0800 Subject: [PATCH] Update test_spmd_debugging.py --- test/spmd/test_spmd_debugging.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/spmd/test_spmd_debugging.py b/test/spmd/test_spmd_debugging.py index 9dd3364641e..8dffe7d7faa 100644 --- a/test/spmd/test_spmd_debugging.py +++ b/test/spmd/test_spmd_debugging.py @@ -55,7 +55,7 @@ def test_debugging_spmd_single_host_tiled(self): color = None text_color = None # console.print(ttable) - fask_table = rich.table.Table(show_header=False, show_lines=False, padding=0, highlight=False, pad_edge=False, box=rich.box.SQUARE) + fask_table = rich.table.Table(show_header=False, show_lines=True, padding=0, highlight=True, pad_edge=False, box=rich.box.SQUARE) col = [] col.append(rich.padding.Padding(rich.align.Align('TPU 0', "center", vertical="middle"), (2,1,2,1), style=rich.style.Style(bgcolor=color, color=text_color))) col.append(rich.padding.Padding(rich.align.Align('TPU 1', "center", vertical="middle"), (2,1,2,1), style=rich.style.Style(bgcolor=color, color=text_color))) @@ -115,7 +115,7 @@ def test_single_host_partial_replication(self): # console.print(ttable) color = None text_color = None - fask_table = rich.table.Table(show_header=False, show_lines=False, padding=0, highlight=False, pad_edge=False, box=rich.box.SQUARE) + fask_table = rich.table.Table(show_header=False, show_lines=True, padding=0, highlight=True, pad_edge=False, box=rich.box.SQUARE) col = [] col.append(rich.padding.Padding(rich.align.Align('TPU [0, 1, 2, 3]', "center", vertical="middle"), (2,0,2,0), style=rich.style.Style(bgcolor=color, color=text_color))) fask_table.add_row(*col) @@ -167,7 +167,7 @@ def test_single_host_replicated(self): # console.print(ttable) color = None text_color = None - fask_table = rich.table.Table(show_header=False, show_lines=False, padding=0, highlight=False, pad_edge=False, box=rich.box.SQUARE) + fask_table = rich.table.Table(show_header=False, show_lines=True, padding=0, highlight=True, pad_edge=False, box=rich.box.SQUARE) col = [] col.append(rich.padding.Padding(rich.align.Align('TPU [0, 1, 2, 3, 4, 5, 6, 7]', "center", vertical="middle"), (0,0,1,0), style=rich.style.Style(bgcolor=color, color=text_color))) fask_table.add_row(*col)