Update test_spmd_debugging.py

pytorch · Nov 22, 2023 · 8e780d3 · 8e780d3
1 parent a5087f3
commit 8e780d3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)