Comments

Author: ColCarroll

pymc3/model_graph.py

@@ -38,7 +38,7 @@ def _get_inputs(self, var, func):
         if parents != upstream:
             det_map = {}
             for d in deterministics:
-                d_set = set([j for j in inputs([func], blockers=[d])])
+                d_set = {j for j in inputs([func], blockers=[d])}
                 if upstream - d_set:
                     det_map[d] = d_set
             for d, d_set in det_map.items():
@@ -171,4 +171,4 @@ def model_to_graphviz(model=None):
     for more information.
     """
     model = pm.modelcontext(model)
-    return ModelGraph(model).make_graph()
+    return ModelGraph(model).make_graph()