Remove deprecated pymc.typing module after moving aliases to pymc.util

Author: aman-coder03

pymc/backends/arviz.py

@@ -40,7 +40,7 @@
 import pymc
 
 from pymc.model import modelcontext
-from pymc.typing import StrongCoords
+from pymc.util import StrongCoords 
 
 if TYPE_CHECKING:
     from pymc.backends.base import MultiTrace

pymc/model/core.py

@@ -65,8 +65,10 @@
     join_nonshared_inputs,
     rewrite_pregrad,
 )
-from pymc.typing import Coords, CoordValue, StrongCoords
 from pymc.util import (
+    Coords,
+    CoordValue,
+    StrongCoords,
     UNSET,
     WithMemoization,
     _UnsetType,
@@ -76,6 +78,7 @@
     treedict,
     treelist,
 )
+
 from pymc.vartypes import continuous_types, discrete_types, typefilter
 
 __all__ = [

pymc/typing.py

@@ -18,7 +18,7 @@
 from collections import namedtuple
 from collections.abc import Sequence
 from copy import deepcopy
-from typing import cast
+from typing import Mapping, TypeAlias, Hashable, cast
 
 import arviz
 import cloudpickle
@@ -31,6 +31,16 @@
 
 from pymc.exceptions import BlockModelAccessError
 
+#Coordinate & Shape Typing
+CoordValue: TypeAlias = Sequence[Hashable] | np.ndarray | None
+Coords: TypeAlias = Mapping[str, CoordValue]
+
+StrongCoordValue: TypeAlias = tuple[Hashable, ...] | None
+StrongCoords: TypeAlias = Mapping[str, StrongCoordValue]
+
+StrongDims: TypeAlias = tuple[str, ...]
+StrongShape: TypeAlias = tuple[int, ...]
+
 
 class _UnsetType:
     """Type for the `UNSET` object to make it look nice in `help(...)` outputs."""