Error in ann_viz: unexpected keyword argument 'name'

[alwkynew]

Hello there,

I am getting the following error, from running ann_viz(model), with even the most basic example.

TypeError: subgraph() got an unexpected keyword argument 'name'

It marks as being from the following code in the library:

~\AppData\Local\Continuum\anaconda3\envs\ml_fuji\lib\site-packages\ann_visualizer\visualize.py in ann_viz(model, view, filename, title)
     88         g.graph_attr.update(splines="false", nodesep='1', ranksep='2');
     89         #Input Layer
---> 90         with g.subgraph(name='cluster_input') as c:
     91             if(type(model.layers[0]) == keras.layers.core.Dense):
     92                 the_label = title+'\n\n\n\nInput Layer';

Any thoughts on why I am getting this?

Thank you.