examples/dataflow: only import nx when needed

diff --git a/examples/dataflow.py b/examples/dataflow.py
index 5c3d5fc0e268e33767db2ec0d27a84e9a33d7cd1..f6e786778b4d3c5699df379321d7cad489546b12 100644 (file)
--- a/examples/dataflow.py
+++ b/examples/dataflow.py
@@ -1,5 +1,4 @@
 import sys
-import networkx as nx
 
 from migen.fhdl import verilog
 from migen.flow.ala import *
@@ -19,5 +18,6 @@ print(verilog.convert(frag))
 
 if len(sys.argv) > 1 and sys.argv[1] == "draw":
        import matplotlib.pyplot as plt
+       import networkx as nx
        nx.draw(g)
        plt.show()