From 0b2787a86b53094fe46a2d98fcd3cc335711a8b0 Mon Sep 17 00:00:00 2001 From: Scott Sanderson Date: Tue, 10 Nov 2015 16:34:53 -0500 Subject: [PATCH] MAINT: Add show_graph to Pipeline. --- zipline/pipeline/pipeline.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/zipline/pipeline/pipeline.py b/zipline/pipeline/pipeline.py index dbd627aa..b5173d5e 100644 --- a/zipline/pipeline/pipeline.py +++ b/zipline/pipeline/pipeline.py @@ -1,6 +1,6 @@ from zipline.utils.input_validation import expect_types, optional -from .term import Term +from .term import Term, AssetExists from .filters import Filter from .graph import TermGraph @@ -145,3 +145,20 @@ class Pipeline(object): columns[screen_name] = screen return TermGraph(columns) + + def show_graph(self, format='svg'): + """ + Render this Pipeline as a DAG. + + Parameters + ---------- + format : str, optional + Image format to render with. Default is 'svg'. + """ + g = self.to_graph('', AssetExists()) + if format == 'svg': + return g.svg + elif format == 'png': + return g.png + else: + return g.jpeg