handle changed API of theano.gof.graph.stack_search

Spaak · Spaak · commit 6238bff6c38f · 2021-01-04T10:21:09.000+01:00
diff --git a/pymc3/model_graph.py b/pymc3/model_graph.py
@@ -69,7 +69,7 @@ def _expand(node) -> Optional[Iterator[Tensor]]:
             else:
                 return None
 
-        stack_search(start=deque([func]), expand=_expand, mode="bfs")
+        stack_search(nodes=deque([func]), expand=_expand, bfs=True)
         return retval
 
     def _filter_parents(self, var, parents) -> Set[VarName]:
