Fix plot to group by array on y-axis

cubed-dev · Jul 13, 2023 · d31fb74 · d31fb74
1 parent c376a18
commit d31fb74
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cubed/extensions/timeline.py b/cubed/extensions/timeline.py
@@ -31,6 +31,10 @@ def on_compute_end(self, dag):
 def create_timeline(stats, start_tstamp, end_tstamp, dst=None):
     stats_df = pd.DataFrame(stats)
 
+    stats_df = stats_df.sort_values(
+        by=["task_create_tstamp", "array_name"], ascending=True
+    )
+
     total_calls = len(stats_df)
 
     palette = sns.color_palette("deep", 6)