Release v2.4

Dicksonchin93 · Apr 23, 2018 · 0248a8d · 0248a8d
1 parent e53e463
commit 0248a8d
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -24,6 +24,12 @@ Use the following command:
 pip install ann_visualizer
 ```
 
+Make sure you have graphviz installed. Install it using:
+
+```bash
+sudo apt-get install graphviz
+```
+
 ## Usage
 
 ```python

diff --git a/ann_visualizer.egg-info/PKG-INFO b/ann_visualizer.egg-info/PKG-INFO
@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: ann-visualizer
-Version: 2.3
+Version: 2.4
 Summary: A python library for visualizing Neural Networks
 Home-page: https://github.com/Prodicode/ann-visualizer
 Author: Tudor Gheorghiu

diff --git a/ann_visualizer/visualize.py b/ann_visualizer/visualize.py
@@ -85,7 +85,7 @@ def ann_viz(model, view=True, filename="network.gv", title="My Neural Network"):
 
         g = Digraph('g', filename=filename);
         n = 0;
-        g.attr(splines="false", nodesep='1', ranksep='2');
+        g.graph_attr.update(splines="false", nodesep='1', ranksep='2');
         #Input Layer
         with g.subgraph(name='cluster_input') as c:
             if(type(model.layers[0]) == keras.layers.core.Dense):

diff --git a/dist/ann_visualizer-2.3.tar.gz b/dist/ann_visualizer-2.3.tar.gz
diff --git a/dist/ann_visualizer-2.4.tar.gz b/dist/ann_visualizer-2.4.tar.gz
diff --git a/setup.py b/setup.py
@@ -25,7 +25,7 @@
 setup(
   name = 'ann_visualizer',
   packages = ['ann_visualizer'],
-  version = '2.3',
+  version = '2.4',
   license="MIT",
   description = 'A python library for visualizing Neural Networks',
   long_description=long_description,