TYP: fix lazy loading and reloading once again

Author: ngoldbaum

numpy/_typing/_array_like.py

@@ -2,7 +2,7 @@
 
 import sys
 from collections.abc import Collection, Callable, Sequence
-from typing import Any, Protocol, TypeAlias, TypeVar, runtime_checkable
+from typing import Any, Protocol, TypeAlias, TypeVar, runtime_checkable, TYPE_CHECKING
 
 import numpy as np
 from numpy import (
@@ -24,6 +24,13 @@
 from ._nested_sequence import _NestedSequence
 from ._shape import _Shape
 
+if TYPE_CHECKING:
+    StringDType = np.dtypes.StringDType
+else:
+    # at runtime outside of type checking importing this from numpy.dtypes
+    # would lead to a circular import
+    from numpy._core.multiarray import StringDType
+
 _T = TypeVar("_T")
 _ScalarType = TypeVar("_ScalarType", bound=generic)
 _ScalarType_co = TypeVar("_ScalarType_co", bound=generic, covariant=True)
@@ -149,7 +156,7 @@ def __array_function__(
     bytes,
 ]
 _ArrayLikeString_co: TypeAlias = _DualArrayLike[
-    np.dtypes.StringDType,
+    StringDType,
     str
 ]
 _ArrayLikeAnyString_co: TypeAlias = _ArrayLikeStr_co | _ArrayLikeBytes_co | _ArrayLikeString_co