Fix ruff formatting, imports, and dev requirements

Author: aman-coder03

pymc/backends/arviz.py

@@ -21,14 +21,11 @@
     TYPE_CHECKING,
     Any,
     Optional,
+    TypeAlias,
     Union,
     cast,
-    TypeAlias,
 )
 
-from pymc.distributions.shape_utils import StrongCoords
-
-
 import numpy as np
 import xarray
 
@@ -42,6 +39,7 @@
 
 import pymc
 
+from pymc.distributions.shape_utils import StrongCoords
 from pymc.model import Model, modelcontext
 from pymc.progress_bar import CustomProgress, default_progress_theme
 from pymc.pytensorf import PointFunc, extract_obs_data
@@ -128,8 +126,9 @@ def find_constants(model: "Model") -> dict[str, Var]:
     return constant_data
 
 
-def coords_and_dims_for_inferencedata(model: Model,) -> tuple[StrongCoords, DimsDict]:
-
+def coords_and_dims_for_inferencedata(
+    model: Model,
+) -> tuple[StrongCoords, DimsDict]:
     """Parse PyMC model coords and dims format to one accepted by InferenceData."""
     coords = {
         cname: np.array(cvals) if isinstance(cvals, tuple) else cvals

pymc/distributions/shape_utils.py

@@ -16,7 +16,7 @@
 
 import warnings
 
-from collections.abc import Sequence
+from collections.abc import Hashable, Mapping, Sequence
 from functools import singledispatch
 from types import EllipsisType
 from typing import Any, TypeAlias, cast
@@ -97,9 +97,6 @@ def _check_shape_type(shape):
 StrongDimsWithEllipsis: TypeAlias = Sequence[str | EllipsisType]
 StrongSize: TypeAlias = TensorVariable | tuple[int | Variable, ...]
 
-from collections.abc import Mapping
-from typing import Hashable
-
 CoordValue: TypeAlias = Sequence[Hashable] | np.ndarray | None
 Coords: TypeAlias = Mapping[str, CoordValue]
 

pymc/model/core.py

@@ -20,8 +20,6 @@
 import warnings
 
 from collections.abc import Iterable, Sequence
-from pymc.distributions.shape_utils import Coords, StrongCoords, CoordValue
-
 from typing import (
     Literal,
     cast,
@@ -46,6 +44,7 @@
 
 from pymc.blocking import DictToArrayBijection, RaveledVars
 from pymc.data import MinibatchOp, is_valid_observed
+from pymc.distributions.shape_utils import Coords, CoordValue, StrongCoords
 from pymc.exceptions import (
     BlockModelAccessError,
     ImputationWarning,
@@ -911,7 +910,7 @@ def unobserved_RVs(self):
     @property
     def coords(self) -> StrongCoords:
         """Coordinate values for model dimensions."""
-        return self._coords 
+        return self._coords
 
     @property
     def dim_lengths(self) -> dict[str, TensorVariable]: