Fix import error in _typing.py

pydata · Feb 2, 2024 · cda1b26 · cda1b26
1 parent 01c3d24
commit cda1b26
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/xarray/namedarray/_typing.py b/xarray/namedarray/_typing.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import typing
 from collections.abc import Hashable, Iterable, Mapping, Sequence
 from enum import Enum
 from types import ModuleType
@@ -19,12 +20,13 @@
 import numpy as np
 from numpy.typing import NDArray
 
-try:
-    from dask.array.core import Array as DaskArray
-    from dask.typing import DaskCollection
-except ImportError:
-    DaskArray = NDArray  # type: ignore
-    DaskCollection: Any = NDArray  # type: ignore
+if typing.TYPE_CHECKING:
+    try:
+        from dask.array.core import Array as DaskArray
+        from dask.typing import DaskCollection
+    except ImportError:
+        DaskArray = NDArray  # type: ignore
+        DaskCollection: Any = NDArray  # type: ignore
 
 
 # Singleton type, as per https://github.com/python/typing/pull/240