tentative fix typechecking

yt/_maintenance/numpy2_compat.py

@@ -1,12 +1,9 @@
 # avoid deprecation warnings in numpy >= 2.0
 
-from importlib.metadata import version
+import numpy as np
 
-from packaging.version import Version
-
-NUMPY_VERSION = Version(version("numpy"))
-
-if NUMPY_VERSION >= Version("2.0.0dev0"):
-    from numpy import trapezoid as trapezoid  # type: ignore [attr-defined]
+if hasattr(np, "trapezoid"):
+    # np.trapz is deprecated in numpy 2.0 in favor of np.trapezoid
+    trapezoid = np.trapezoid
 else:
-    from numpy import trapz as trapezoid  # type: ignore [attr-defined] # noqa: F401
+    trapezoid = np.trapz  # noqa: NPY201

yt/_typing.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional
+from typing import Any, Optional, TypeAlias
 
 import numpy as np
 import unyt as un
@@ -25,7 +25,7 @@
 AxisOrder = tuple[AxisName, AxisName, AxisName]
 
 # types that can be converted to un.Unit
-Unit = un.Unit | str
+Unit: TypeAlias = un.Unit | str
 
 # types that can be converted to un.unyt_quantity
 Quantity = un.unyt_quantity | tuple[float, Unit]