Filter deterministics from initial_points returned by init_nuts

Author: ricardoV94

pymc/sampling.py

@@ -2547,4 +2547,11 @@ def init_nuts(
 
     step = pm.NUTS(potential=potential, model=model, **kwargs)
 
+    # Filter deterministics from initial_points
+    value_var_names = [var.name for var in model.value_vars]
+    initial_points = [
+        {k: v for k, v in initial_point.items() if k in value_var_names}
+        for initial_point in initial_points
+    ]
+
     return initial_points, step

pymc/tests/test_sampling.py

@@ -953,14 +953,12 @@ def check_exec_nuts_init(method):
         assert isinstance(start, list)
         assert len(start) == 1
         assert isinstance(start[0], dict)
-        assert model.a.tag.value_var.name in start[0]
-        assert model.b.tag.value_var.name in start[0]
+        assert set(start[0].keys()) == {v.name for v in model.value_vars}
         start, _ = pm.init_nuts(init=method, n_init=10, chains=2, seeds=[1, 2])
         assert isinstance(start, list)
         assert len(start) == 2
         assert isinstance(start[0], dict)
-        assert model.a.tag.value_var.name in start[0]
-        assert model.b.tag.value_var.name in start[0]
+        assert set(start[0].keys()) == {v.name for v in model.value_vars}
 
 
 @pytest.mark.parametrize(