ImportError: cannot import name 'normalize_axis_index' from 'pytensor.npy_2_compat'

[twiecki]

pixi.toml:

[workspace]
authors = ["Thomas Wiecki <thomas.wiecki@gmail.com>"]
channels = ["conda-forge"]
name = "ias"
platforms = ["osx-arm64"]
version = "0.1.0"

[tasks]

[dependencies]
pymc-marketing = ">=0.16.0,<0.17"
marimo = ">=0.16.5,<0.17"
fastmcp = ">=2.12.4,<3"

>>pixi run python -c "import pytensor; print(pytensor.__version__)"
2.35.0

import pymc_marketing in marimo produces:

Traceback (most recent call last):
  Cell 
marimo:///Users/twiecki/projects/insight-agent-skills/notebook.py#cell=cell-0

, line 1, in <module>
    import pymc_marketing
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/__init__.py", line 16, in <module>
    from pymc_marketing import clv, customer_choice, mmm
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/customer_choice/__init__.py", line 16, in <module>
    from pymc_marketing.customer_choice.mv_its import (
    ...<2 lines>...
    )
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/customer_choice/mv_its.py", line 29, in <module>
    from pymc_marketing.mmm.additive_effect import MuEffect
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/mmm/__init__.py", line 16, in <module>
    from pymc_marketing.mmm import base, mmm, preprocessing, validating
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/mmm/mmm.py", line 38, in <module>
    from pymc_marketing.mmm.components.adstock import (
    ...<2 lines>...
    )
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/mmm/components/adstock.py", line 70, in <module>
    from pymc_marketing.mmm.transformers import (
    ...<5 lines>...
    )
  File "/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pymc_marketing/mmm/transformers.py", line 24, in <module>
    from pytensor.npy_2_compat import normalize_axis_index
ImportError: cannot import name 'normalize_axis_index' from 'pytensor.npy_2_compat' (/Users/twiecki/projects/insight-agent-skills/.pixi/envs/default/lib/python3.13/site-packages/pytensor/npy_2_compat.py)
Launch debugger
Get help

[twiecki]

Not sure if it's a known pytensor deprecation, so I also opened an issue here: pymc-devs/pytensor#1675

[williambdean]

Seems to be solved with #2011