Adding a vis for pipeline using GraphViz and an example is shown

Author: raghukiran1224

codeflare/pipelines/utils.py

@@ -0,0 +1,20 @@
+import graphviz
+import codeflare.pipelines.Datamodel as dm
+
+
+def pipeline_to_graph(pipeline: dm.Pipeline) -> graphviz.Digraph:
+    """
+    Converts the given pipeline to a networkX graph for visualization.
+
+    :param pipeline: Pipeline to convert to networkX graph
+    :return: A directed graph representing this pipeline
+    """
+    graph = graphviz.Digraph()
+    pipeline_nodes = pipeline.get_nodes()
+    for pre_node in pipeline_nodes.values():
+        post_nodes = pipeline.get_post_nodes(pre_node)
+        graph.node(pre_node.get_node_name())
+        for post_node in post_nodes:
+            graph.node(post_node.get_node_name())
+            graph.edge(pre_node.get_node_name(), post_node.get_node_name())
+    return graph

codeflare_pipelines.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ codeflare/pipelines/Datamodel.py
 codeflare/pipelines/Exceptions.py
 codeflare/pipelines/Runtime.py
 codeflare/pipelines/__init__.py
+codeflare/pipelines/utils.py
 codeflare_pipelines.egg-info/PKG-INFO
 codeflare_pipelines.egg-info/SOURCES.txt
 codeflare_pipelines.egg-info/dependency_links.txt