From 26cef8d959d243f577639144cd886b31e2336f90 Mon Sep 17 00:00:00 2001 From: Scott Sanderson Date: Fri, 20 Nov 2015 21:09:12 -0500 Subject: [PATCH] DOC: Better doc and error messages for show_graph. --- zipline/pipeline/pipeline.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/zipline/pipeline/pipeline.py b/zipline/pipeline/pipeline.py index b5173d5e..7bfaa4b1 100644 --- a/zipline/pipeline/pipeline.py +++ b/zipline/pipeline/pipeline.py @@ -152,7 +152,7 @@ class Pipeline(object): Parameters ---------- - format : str, optional + format : {'svg', 'png', 'jpeg'} Image format to render with. Default is 'svg'. """ g = self.to_graph('', AssetExists()) @@ -160,5 +160,7 @@ class Pipeline(object): return g.svg elif format == 'png': return g.png - else: + elif format == 'jpeg': return g.jpeg + else: + raise ValueError("Unknown graph format %r." % format)