From be83bc911457083eaa08701495d6832443c7d116 Mon Sep 17 00:00:00 2001 From: Angus Hollands Date: Wed, 18 Dec 2024 15:36:26 +0000 Subject: [PATCH] fix: appease mypy --- src/dask_awkward/layers/layers.py | 2 +- src/dask_awkward/lib/core.py | 2 +- src/dask_awkward/lib/optimize.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/dask_awkward/layers/layers.py b/src/dask_awkward/layers/layers.py index 5b360400..a3348a5b 100644 --- a/src/dask_awkward/layers/layers.py +++ b/src/dask_awkward/layers/layers.py @@ -15,7 +15,7 @@ _dask_uses_tasks = hasattr(dask.blockwise, "Task") if _dask_uses_tasks: - from dask.blockwise import Task, TaskRef # type: ignore + from dask.blockwise import Task, TaskRef if TYPE_CHECKING: from awkward import Array as AwkwardArray diff --git a/src/dask_awkward/lib/core.py b/src/dask_awkward/lib/core.py index 6becf1b1..9838d625 100644 --- a/src/dask_awkward/lib/core.py +++ b/src/dask_awkward/lib/core.py @@ -62,7 +62,7 @@ ) if _dask_uses_tasks: - from dask.blockwise import TaskRef # type: ignore + from dask.blockwise import TaskRef if TYPE_CHECKING: from awkward.contents.content import Content diff --git a/src/dask_awkward/lib/optimize.py b/src/dask_awkward/lib/optimize.py index 97724fa5..cb884320 100644 --- a/src/dask_awkward/lib/optimize.py +++ b/src/dask_awkward/lib/optimize.py @@ -23,7 +23,7 @@ from dask_awkward.utils import first if _dask_uses_tasks: - from dask.blockwise import GraphNode, Task, TaskRef # type: ignore + from dask.blockwise import GraphNode, Task, TaskRef if TYPE_CHECKING: from awkward._nplikes.typetracer import TypeTracerReport