def plot(self, output=None): import matplotlib.pyplot as plt #pylint:disable=import-error plt.close() networkx.draw(self.make_graph()) if output: plt.savefig(output) else: plt.show()