Add vars property to CompoundStep

pymc-devs · Sep 22, 2021 · d926746 · d926746
1 parent 31a9f14
commit d926746
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/pymc3/sampling.py b/pymc3/sampling.py
@@ -187,11 +187,7 @@ def assign_step_methods(model, step=None, methods=STEP_METHODS, step_kwargs=None
         except TypeError:
             steps.append(step)
         for step in steps:
-            try:
-                assigned_vars = assigned_vars.union(set(step.vars))
-            except AttributeError:
-                for method in step.methods:
-                    assigned_vars = assigned_vars.union(set(method.vars))
+            assigned_vars = assigned_vars.union(set(step.vars))
 
     # Use competence classmethods to select step methods for remaining
     # variables

diff --git a/pymc3/step_methods/compound.py b/pymc3/step_methods/compound.py
@@ -71,3 +71,7 @@ def reset_tuning(self):
         for method in self.methods:
             if hasattr(method, "reset_tuning"):
                 method.reset_tuning()
+
+    @property
+    def vars(self):
+        return [var for method in self.methods for var in method.vars]