Skip to content

Commit

Permalink
improve warning messages
Browse files Browse the repository at this point in the history
  • Loading branch information
dimitri-yatsenko committed Feb 8, 2019
1 parent fb14029 commit 90cf697
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions datajoint/erd.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ class ERD:
"""

def __init__(self, *args, **kwargs):
warnings.warn('ERD functionality depends on matplotlib, networkx, and pygraphviz. '
warnings.warn('ERD functionality depends on matplotlib and pygraphviz. '
'Please install both of these libraries to enable the ERD feature.')
else:
class ERD(nx.DiGraph):
Expand Down Expand Up @@ -228,7 +228,6 @@ def _make_graph(self):
return graph

def make_dot(self):
import networkx as nx

graph = self._make_graph()
graph.nodes()
Expand Down

0 comments on commit 90cf697

Please sign in to comment.