improve

Signed-off-by: Yicheng-Lu-llll <luyc58576@gmail.com>
flyteorg · Apr 24, 2023 · ec9df69 · ec9df69
1 parent f7ea236
commit ec9df69
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/flytekit/deck/deck.py b/flytekit/deck/deck.py
@@ -121,6 +121,7 @@ def html(self) -> str:
         # set the accuracy to microsecond
         df["ProcessTime"] = df["ProcessTime"].apply(lambda time: "{:.6f}".format(time))
         df["WallTime"] = df["WallTime"].apply(lambda time: "{:.6f}".format(time))
+
         width = 1400
         gantt_chart_html = GanttChartRenderer().to_html(df, chart_width=width)
         time_table_html = TableRenderer().to_html(

diff --git a/tests/flytekit/unit/deck/test_deck.py b/tests/flytekit/unit/deck/test_deck.py
@@ -37,7 +37,7 @@ def test_timeline_deck():
     ctx.user_space_params._decks = []
     timeline_deck = ctx.user_space_params.timeline_deck
     timeline_deck.append_time_info(time_info)
-    assert timeline_deck.name == "time line"
+    assert timeline_deck.name == "Timeline"
     assert len(timeline_deck.time_info) == 1
     assert timeline_deck.time_info[0] == time_info
     assert len(ctx.user_space_params.decks) == 1