fix: Fix random variables with missing values in pymc deterministics

Author: aseyboldt

python/nutpie/compile_pymc.py

@@ -202,6 +202,7 @@ def compile_pymc_model(model: "pm.Model", **kwargs) -> CompiledPyMCModel:
             raise ValueError(f"Shared variables must have unique names: {val.name}")
         shared_data[val.name] = val.get_value().copy()
         shared_vars[val.name] = val
+        seen.add(val)
 
     for val in shared_data.values():
         val.flags.writeable = False

tests/test_pymc.py

@@ -99,7 +99,8 @@ def test_pymc_model_shared():
 def test_missing():
     with pm.Model(coords={"obs": range(4)}) as model:
         mu = pm.Normal("mu")
-        pm.Normal("y", mu, observed=[0, -1, 1, np.nan], dims="obs")
+        y = pm.Normal("y", mu, observed=[0, -1, 1, np.nan], dims="obs")
+        pm.Deterministic("y2", 2 * y, dims="obs")
 
     compiled = nutpie.compile_pymc_model(model)
     tr = nutpie.sample(compiled, chains=1, seed=1)