take _task_spec classes from blockwise

dask-contrib · Dec 17, 2024 · 49a6d62 · 49a6d62
1 parent 64b2a66
commit 49a6d62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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