Avoid runtime broadcast error due to dot_to_mul rewrite

[ricardoV94]

For convenience also allow Ellipsis in specify_shape.

Closes #1782

[ricardoV94]

@tomicapretto or @ivanistheone, could either of you check if the patch fixes the original problem?

You may need to apply the changes locally because there are probably other major breaking changes in pytensor main related to pymc/bambi

[tomicapretto]

@ricardoV94 yes, I'll reply in a moment

[ivanistheone]

could either of you check if the patch fixes the original problem?

Everything worked fine after when latest pymc, bambi and pip install -e pytensor from the branch dot_to_mul_bug.

You may need to apply the changes locally because there are probably other major breaking changes in pytensor main related to pymc/bambi

I had to add from pytensor.sparse.math import sp_sum to the top of pytensor/sparse/basic.py because pymc was looking for it there.

[ricardoV94]

Great

[tomicapretto]

I confirm I can run the original example, shared in the issue, without problem (after adjusting one import)..
Thanks @ricardoV94 !