Merge pull request #1 from tsmanner/dot_serializer

Dot serializer
Vipyr · Jun 9, 2019 · ebb2586 · ebb2586
2 parents 3489c3b + f25e561
commit ebb2586
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/bazaarci/runner/__init__.py b/bazaarci/runner/__init__.py
@@ -1,3 +1,4 @@
+from .node import Node
 from .graph import Graph
 from .product import Product
 from .step import Step

diff --git a/bazaarci/serializers/dot.py b/bazaarci/serializers/dot.py
@@ -66,7 +66,9 @@ def product_to_dot(product):
     color = "green" if product.is_set() else "yellow"
     return f"\"{product.name}\" [style=filled color={color}]"
 
-apply_serializer(Graph, "to_dot", graph_to_dot)
-apply_serializer(Step, "to_dot", step_to_dot)
-apply_serializer(SubprocessStep, "to_dot", subprocess_step_to_dot)
-apply_serializer(Product, "to_dot", product_to_dot)
+
+def initialize():
+    apply_serializer(Graph, "to_dot", graph_to_dot)
+    apply_serializer(Step, "to_dot", step_to_dot)
+    apply_serializer(SubprocessStep, "to_dot", subprocess_step_to_dot)
+    apply_serializer(Product, "to_dot", product_to_dot)