From 49a6d62fe2135a1832398abdccb60bd4b96793ed Mon Sep 17 00:00:00 2001
From: Lindsey Gray <lindsey.gray@gmail.com>
Date: Tue, 17 Dec 2024 12:47:11 -0600
Subject: [PATCH] take _task_spec classes from blockwise

---
 src/dask_awkward/lib/optimize.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dask_awkward/lib/optimize.py b/src/dask_awkward/lib/optimize.py
index 4b9dd6cf..0eba92fa 100644
--- a/src/dask_awkward/lib/optimize.py
+++ b/src/dask_awkward/lib/optimize.py
@@ -22,7 +22,7 @@
 from dask_awkward.utils import first
 
 if _dask_uses_tasks:
-    from dask._task_spec import GraphNode, Task, TaskRef
+    from dask.blockwise import GraphNode, Task, TaskRef
 
 if TYPE_CHECKING:
     from awkward._nplikes.typetracer import TypeTracerReport