From ff174c1a45dda9d809144464978a7709e69c674b Mon Sep 17 00:00:00 2001
From: J Faleiro <j@falei.ro>
Date: Thu, 21 Feb 2019 13:54:23 -0500
Subject: [PATCH] support for Jupyter inline plots Jupyter

---
 ann_visualizer/visualize.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ann_visualizer/visualize.py b/ann_visualizer/visualize.py
index 72a9a7c..b55d714 100644
--- a/ann_visualizer/visualize.py
+++ b/ann_visualizer/visualize.py
@@ -204,3 +204,4 @@ def ann_viz(model, view=True, filename="network.gv", title="My Neural Network"):
         g.edge_attr.update(arrowhead="none", color="#707070");
         if view == True:
             g.view();
+        return g