Skip to content

MarginalModel.unmarginalize doesn't accept var_names #309

Open
@ricardoV94

Description

@ricardoV94
import pymc as pm
from pymc_experimental import MarginalModel

with MarginalModel() as m:
    x = pm.Bernoulli("x", p=0.5)
    y = pm.Normal("y", x)
    
m.marginalize(["x"])
m.unmarginalize(["x"])  # ValueError: list.remove(x): x not in list

m.unmarginalize([x]) is fine, but this is problematic because x may not be accessible to the caller

CC @zaxtax

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions