From d31fb74500c158914b866f3f9ac9c29ae91c86a2 Mon Sep 17 00:00:00 2001
From: Tom White <tom.e.white@gmail.com>
Date: Thu, 13 Jul 2023 17:19:35 +0100
Subject: [PATCH] Fix plot to group by array on y-axis

---
 cubed/extensions/timeline.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cubed/extensions/timeline.py b/cubed/extensions/timeline.py
index 1b8a269f..310379d9 100644
--- 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)