Skip to content

Arange with constant outputs does not result into a type with constant shape #445

Open
@ferrine

Description

@ferrine

Describe the issue:

pt.arange allocates a vector of length that depends on input values but the output shape has dynamic type

Reproducable code example:

assert pt.arange(10).type.shape == (10, )

Error message:

(None, ) != (10,)

PyTensor version information:

'2.16.1'

Context for the issue:

used here https://github.com/pymc-labs/pymc-marketing/blob/main/pymc_marketing/mmm/transformers.py#L102C50-L102C59

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions