Fix _check_start_shape

pymc-devs · May 14, 2021 · 77d6c60 · 77d6c60
1 parent a0e36d0
commit 77d6c60
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pymc3/sampling.py b/pymc3/sampling.py
@@ -684,6 +684,11 @@ def sample(
 def _check_start_shape(model, start):
     if not isinstance(start, dict):
         raise TypeError("start argument must be a dict or an array-like of dicts")
+
+    # Filter "non-input" variables
+    initial_point = model.initial_point
+    start = {k: v for k, v in deepcopy(start).items() if k in initial_point}
+
     e = ""
     for var in model.basic_RVs:
         var_shape = model.fastfn(var.shape)(start)