Merge pull request #14 from Jsoconno/hotfix-flow

fixed issue with flow logic
jsoconno · Jan 6, 2021 · 014989d · 014989d
2 parents de505c6 + bcf8344
commit 014989d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/architectures/core/__init__.py b/architectures/core/__init__.py
@@ -451,7 +451,7 @@ def __init__(self, nodes, **attrs):
                         cluster = self.start_node
                         self.start_node = get_cluster_node(self.start_node)
                         self.edge_attrs.update({"ltail": cluster.name})
-                    elif isinstance(current_start_node, (Cluster, Group)) and isinstance(current_end_node, (Cluster, Group)):
+                    elif isinstance(self.start_node, (Cluster, Group)) and isinstance(self.end_node, (Cluster, Group)):
                         start_cluster = self.start_node
                         end_cluster = self.end_node
                         self.start_node = get_cluster_node(self.start_node)

diff --git a/my-architecture.png b/my-architecture.png
diff --git a/tests/simple-test.py b/tests/simple-test.py
@@ -15,4 +15,4 @@
     with Cluster("Cluster C") as cluster_c:
         g = Node("G")
 
-    Flow([a, c, cluster_b, g, f, b], color="goldenrod")
+    Flow([a, c, cluster_b, cluster_a, g, f, b], color="goldenrod")