Type-check guard numpy.typing import

Numpy versions less than 1.20 do not have the typing module.
equinor · Nov 10, 2023 · 850609f · 850609f
1 parent 0d3ee8e
commit 850609f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/resfo/types.py b/src/resfo/types.py
@@ -23,10 +23,12 @@
 
 """
 import warnings
-from typing import Union
+from typing import TYPE_CHECKING, Union
 
 import numpy as np
-from numpy.typing import ArrayLike
+
+if TYPE_CHECKING:
+    from numpy.typing import ArrayLike
 
 # np dtype for res types with fixed width
 static_dtypes = {
@@ -48,7 +50,7 @@ class MESS:
 
 
 ReadArrayValue = Union[np.ndarray, MESS]
-WriteArrayValue = Union[ArrayLike, MESS]
+WriteArrayValue = Union["ArrayLike", MESS]
 
 
 def to_np_type(type_keyword):