BUG: ImportError: cannot import name 'batched_dot' from 'pytensor.tensor.blas'

[noirchen]

Describe the issue:

I installed fresh pymc via conda on mac

conda create -c conda-forge -n pymc_env "pymc>=6"
conda activate pymc_env

and when I import pytensor, I got

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[1], line 1
----> 1 import pytensor

File ~/.conda/envs/pymc/lib/python3.14/site-packages/pytensor/__init__.py:19
     15 from pytensor.configdefaults import config
     18 # isort: off
---> 19 from pytensor import tensor
     20 from pytensor.compile import (
     21     In,
     22     Mode,
   (...)     25     wrap_py,
     26 )
     27 from pytensor.compile.maker import function

File ~/.conda/envs/pymc/lib/python3.14/site-packages/pytensor/tensor/__init__.py:125
    122 from numpy import e, euler_gamma, inf, nan, newaxis, pi
    124 from pytensor.tensor.basic import *
--> 125 from pytensor.tensor.blas import batched_dot, batched_tensordot
    126 from pytensor.tensor.extra_ops import *
    127 from pytensor.tensor.interpolate import interp, interpolate1d

ImportError: cannot import name 'batched_dot' from 'pytensor.tensor.blas' (/Users/xxx/.conda/envs/pymc/lib/python3.14/site-packages/pytensor/tensor/blas/__init__.py)

Reproducable code example:

import pytensor

Error message:

PyTensor version information:

Info from python -c "import pytensor; print(pytensor.config)"

Details

Traceback (most recent call last): File "", line 1, in import pytensor; print(pytensor.config) ^^^^^^^^^^^^^^^ File "/Users/xxx/.conda/envs/pymc/lib/python3.14/site-packages/pytensor/__init__.py", line 19, in from pytensor import tensor File "/Users/xxx/.conda/envs/pymc/lib/python3.14/site-packages/pytensor/tensor/__init__.py", line 125, in from pytensor.tensor.blas import batched_dot, batched_tensordot ImportError: cannot import name 'batched_dot' from 'pytensor.tensor.blas' (/Users/xxx/.conda/envs/pymc/lib/python3.14/site-packages/pytensor/tensor/blas/__init__.py)

Context for the issue:

No response

[ricardoV94]

what pytensor and pymc versions do you have? sounds like incompatible versions

[noirchen]

@ricardoV94 pytensor 3.0.4, pymc 6.3.1.

[noirchen]

I downgrade to python=3.13.15, pytensor=2.38.2, pymc=5.27.1, and now it works fine