Nutpie fails to sample when RV changes shape following `with_data`

[ricardoV94]

import pymc as pm
import nutpie

with pm.Model() as model:
    x = pm.Data('x', [0,1,2])
    beta = pm.Normal('beta', mu=0, sigma=1)
    y = pm.Normal("y", mu=beta * x, sigma=1)   # fine if it's observed and we also change it's data

compiled_model = nutpie.compile_pymc_model(model)

compiled_model_set = compiled_model.with_data(x=[1,2])
nutpie.sample(compiled_model_set, chains=1)
# RuntimeError: All initialization points failed

# Caused by:
#     Logp function returned error: ErrorCode(1)

[ricardoV94]

And not sure if related, but if I have a Deterministic in the graph it just hangs, doesn't even begin to sample:

import pymc as pm
import nutpie

with pm.Model() as model:
    x = pm.Data('x', [0,1,2])
    y = pm.Data("y", [0,1,2])
    beta = pm.Normal('beta', mu=0, sigma=1)
    obs = pm.Normal("obs", mu=beta * x, sigma=1, observed=y)
    pm.Deterministic("mu", beta * x)  # fine if this is not included
    

compiled_model = nutpie.compile_pymc_model(model)

compiled_model_set = compiled_model.with_data(x=[1,2], y=[1, 2])
nutpie.sample(compiled_model_set, chains=1)  # hangs