Use definition of DTypeLike from Numpy if found (#4941)

* Use definition of DTypeLike from Numpy And fall back to original definition with older versions of numpy * fix flake8
pydata · Feb 23, 2021 · 348eb48 · 348eb48
1 parent fd001f1
commit 348eb48
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/xarray/core/npcompat.py b/xarray/core/npcompat.py
@@ -75,10 +75,12 @@ def moveaxis(a, source, destination):
     return result
 
 
-# Type annotations stubs. See also / to be replaced by:
-# https://github.com/numpy/numpy/issues/7370
-# https://github.com/numpy/numpy-stubs/
-DTypeLike = Union[np.dtype, str]
+# Type annotations stubs.
+try:
+    from numpy.typing import DTypeLike
+except ImportError:
+    # fall back for numpy < 1.20
+    DTypeLike = Union[np.dtype, str]
 
 
 # from dask/array/utils.py