use module_available in nputils as well

pydata · Jan 12, 2024 · c54b7fa · c54b7fa
1 parent 2e4a4c3
commit c54b7fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xarray/core/nputils.py b/xarray/core/nputils.py
@@ -11,9 +11,9 @@
 from xarray.core.utils import module_available
 
 # remove once numpy 2.0 is the oldest supported version
-try:
+if module_available("numpy", minversion="2.0.0.dev0"):
     from numpy.lib.array_utils import normalize_axis_index
-except ImportError:
+else:
     from numpy.core.multiarray import normalize_axis_index  # type: ignore[attr-defined]
 
 # remove once numpy 2.0 is the oldest supported version