From 833831922d4f070003dff9eac1975e999bc22a6e Mon Sep 17 00:00:00 2001
From: Patrick Hoefler <61934744+phofl@users.noreply.github.com>
Date: Tue, 6 Aug 2024 12:32:59 +0200
Subject: [PATCH] Run graph normalisation after dask order (#8818)

---
 distributed/scheduler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/distributed/scheduler.py b/distributed/scheduler.py
index adf99113b9..a7cd91a765 100644
--- a/distributed/scheduler.py
+++ b/distributed/scheduler.py
@@ -4884,6 +4884,7 @@ async def update_graph(
                 internal_priority = await offload(
                     dask.order.order, dsk=dsk, dependencies=stripped_deps
                 )
+            dsk = valmap(_normalize_task, dsk)
 
             self._create_taskstate_from_graph(
                 dsk=dsk,
@@ -9383,5 +9384,4 @@ def _materialize_graph(
         deps.discard(k)
         dependencies[k] = deps
 
-    dsk = valmap(_normalize_task, dsk)
     return dsk, dependencies, annotations_by_type