search

pymc-devs / pymc-experimental

https://pymc-experimental.readthedocs.io
Other
72 stars 46 forks source link

MarginalModel freezes mutable dim lengths #315

Closed ricardoV94 closed 2 months ago

ricardoV94 commented 4 months ago 
import pymc
from pymc_experimental import MarginalModel

with MarginalModel(coords_mutable={"trial": [0, 1, 2]}) as m:
    x = pm.Normal("x", dims="trial")
    print(type(m.dim_lengths["trial"]))  # <class 'pytensor.tensor.sharedvar.ScalarSharedVariable'>
    m.logp()
    print(type(m.dim_lengths["trial"]))  # <class 'pytensor.tensor.variable.TensorConstant'>

This does not happen with pymc Model

ricardoV94 commented 2 months ago

This is no longer relevant because coords are always mutable now