Fix matplotlib import; spacing

Author: vgoliber

Exploring_Pegasus/biclique_embedding.py

@@ -19,9 +19,11 @@
 from minorminer import find_embedding
 from dwave.system.samplers import DWaveSampler
 
-import matplotlib
-matplotlib.use("agg")
-import matplotlib.pyplot as plt
+try:
+ import matplotlib.pyplot as plt
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
 
 # Form the biclique
 N = int(sys.argv[1])

Exploring_Pegasus/clique_embedding.py

@@ -18,9 +18,11 @@
 from minorminer.busclique import find_clique_embedding
 from dwave.system.samplers import DWaveSampler
 
-import matplotlib
-matplotlib.use("agg")
-import matplotlib.pyplot as plt
+try:
+ import matplotlib.pyplot as plt
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
 
 N = int(sys.argv[1])
 G = nx.complete_graph(N)

Pegasus_Embedding_Video/double_plot.py

@@ -12,11 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import matplotlib
-matplotlib.use("agg")
-from matplotlib import pyplot as plt, colors as mpl_color
 import networkx as nx, dwave_networkx as dnx
 
+try:
+ import matplotlib.pyplot as plt
+ import matplotlib.colors as mpl_color
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
+ import matplotlib.colors as mpl_color
+
 def color(i, n):
  r, g, b = mpl_color.hsv_to_rgb((i/n, .25 + (i%4)/4, 1))
  return r, g, b, 1.

Pegasus_Embedding_Video/draw_yield.py

@@ -13,11 +13,14 @@
 # limitations under the License.
 
 from pegasus_graph import P6
-import matplotlib
-matplotlib.use("agg")
-from matplotlib import pyplot as plt
 import dwave_networkx as dnx
 
+try:
+ import matplotlib.pyplot as plt
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
+
 dnx.draw_pegasus_yield(P6, node_size = 80)
 
 dnx.draw_pegasus(P6, node_size = 80)

Pegasus_Embedding_Video/embed_draw_clique.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 from pegasus_graph import P16
-import matplotlib
-matplotlib.use("agg")
-from matplotlib import pyplot as plt
 import networkx as nx, dwave_networkx as dnx
-
 from minorminer import busclique
 
+try:
+ import matplotlib.pyplot as plt
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
+
 clique_cache = busclique.busgraph_cache(P16)
 clique_embedding = clique_cache.largest_clique()
 

Pegasus_Embedding_Video/embed_draw_random.py

@@ -13,11 +13,16 @@
 # limitations under the License.
 
 from pegasus_graph import P6
-import matplotlib
-matplotlib.use("agg")
-from matplotlib import pyplot as plt, colors as mpl_color
 import networkx as nx, dwave_networkx as dnx, minorminer
 
+try:
+ import matplotlib.pyplot as plt
+ import matplotlib.colors as mpl_color
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
+ import matplotlib.colors as mpl_color
+
 nodes = 80
 edge_probability = 0.1
 G = nx.gnp_random_graph(nodes, edge_probability, seed = 0)

Pegasus_Embedding_Video/embed_draw_sparse.py

@@ -14,17 +14,22 @@
 
 from pegasus_graph import P16, P6
 import minorminer.layout as mml, dwave_networkx as dnx, networkx as nx
-import matplotlib
-matplotlib.use("agg")
-from matplotlib import pyplot as plt, colors as mpl_color
+
+try:
+ import matplotlib.pyplot as plt
+ import matplotlib.colors as mpl_color
+except ImportError:
+ matplotlib.use("agg")
+ import matplotlib.pyplot as plt
+ import matplotlib.colors as mpl_color
 
 # Draw a small P6 graph
 n = 200
 C = nx.random_regular_graph(3, n)
 
-emb, (layout_C, layout_P) = mml.find_embedding(C, P6, random_seed = 1,
- return_layouts = True,
- threads = 3)
+emb, (layout_C, layout_P) = mml.find_embedding(C, P6, random_seed=1,
+ return_layouts=True,
+ threads=3)
 
 plt.figure(figsize=(20, 20))
 
@@ -44,9 +49,15 @@
  n = 850
  C = nx.random_regular_graph(3, n)
 
- emb, (layout_C, layout_P) = mml.find_embedding(C, P16, random_seed = 2,
- return_layouts = True, layout = (None, None),
- threads = 3, verbose = 2, interactive = True, tries = 30, max_no_improvement = 10000, timeout = 10000000)
+ emb, (layout_C, layout_P) = mml.find_embedding(C, P16, random_seed=2,
+ return_layouts=True, 
+ layout=(None, None),
+ threads=3, 
+ verbose=2, 
+ interactive=True, 
+ tries=30, 
+ max_no_improvement=10000, 
+ timeout=10000000)
 
  plt.figure(figsize=(20, 20))