Fixes PyTensor compatibility for normalize_axis_index

Handles compatibility issues with PyTensor's `normalize_axis_index` function.

PyTensor 2.35+ no longer exports `normalize_axis_index`, so the code now attempts to import it from `numpy._core.numeric` or `numpy.core.numeric` if the initial import fails.

Author: lfiaschi

pymc_marketing/mmm/transformers.py

@@ -21,7 +21,17 @@
 import pymc as pm
 import pytensor.tensor as pt
 from pymc.distributions.dist_math import check_parameters
-from pytensor.npy_2_compat import normalize_axis_index
+
+# Import normalize_axis_index - handle compatibility
+try:
+    from pytensor.npy_2_compat import normalize_axis_index
+except ImportError:
+    # PyTensor 2.35+ no longer exports this, use numpy directly
+    try:
+        from numpy._core.numeric import normalize_axis_index
+    except (ImportError, AttributeError):
+        # Older numpy versions
+        from numpy.core.numeric import normalize_axis_index
 
 
 class ConvMode(str, Enum):