From 64b2a66d84357c9c42830494a2cd53cee579f8e2 Mon Sep 17 00:00:00 2001
From: Lindsey Gray <lindsey.gray@gmail.com>
Date: Tue, 17 Dec 2024 12:45:46 -0600
Subject: [PATCH] only check blockwise for Task; grab Task, TaskRef from
 blockwise

---
 src/dask_awkward/layers/layers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/dask_awkward/layers/layers.py b/src/dask_awkward/layers/layers.py
index 305409a2..27e1f15d 100644
--- a/src/dask_awkward/layers/layers.py
+++ b/src/dask_awkward/layers/layers.py
@@ -12,10 +12,10 @@
 
 from dask_awkward.utils import LazyInputsDict
 
-_dask_uses_tasks = hasattr(dask, "_task_spec") and hasattr(dask.blockwise, "Task")
+_dask_uses_tasks = hasattr(dask.blockwise, "Task")
 
 if _dask_uses_tasks:
-    from dask._task_spec import Task, TaskRef
+    from dask.blockwise import Task, TaskRef
 
 if TYPE_CHECKING:
     from awkward import Array as AwkwardArray