grab TaskRef from blockwise

dask-contrib · Dec 17, 2024 · dd1e5dd · dd1e5dd
1 parent 49a6d62
commit dd1e5dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dask_awkward/lib/core.py b/src/dask_awkward/lib/core.py
@@ -62,7 +62,7 @@
 )
 
 if _dask_uses_tasks:
-    from dask._task_spec import TaskRef
+    from dask.blockwise import TaskRef # type: ignore
 
 if TYPE_CHECKING:
     from awkward.contents.content import Content