diff --git a/pymc3/model_graph.py b/pymc3/model_graph.py index 512b529ae5..743f4142c0 100644 --- a/pymc3/model_graph.py +++ b/pymc3/model_graph.py @@ -12,7 +12,6 @@ # of variables in a model. RV = Tensor - class ModelGraph: def __init__(self, model): self.model = model @@ -35,10 +34,10 @@ def _get_ancestors(self, var, func) -> MutableSet[RV]: """ # this contains all of the variables in the model EXCEPT var... - vars = set(self.var_list) + vars: MutableSet[RV] = set(self.var_list) vars.remove(var) - - blockers = set() + + blockers: MutableSet[RV] = set() retval = set() def _expand(node) -> Optional[Iterator[Tensor]]: if node in blockers: