equivalent_models

pymc_extras.utils.model_equivalence.equivalent_models(model1: Model, model2: Model, *, strict_dtype: bool = True) → bool

Check whether two PyMC models are equivalent.

Examples

import pymc as pm
from pymc_extras.utils.model_equivalence import equivalent_models

with pm.Model() as m1:
    x = pm.Normal("x")
    y = pm.Normal("y", x)

with pm.Model() as m2:
    x = pm.Normal("x")
    y = pm.Normal("y", x + 1)

with pm.Model() as m3:
    x = pm.Normal("x")
    y = pm.Normal("y", x)

assert not equivalent_models(m1, m2)
assert equivalent_models(m1, m3)