Doctest fix. (#2987)

Adds the fix prescribed in #2925.
quantumlib · May 8, 2020 · f0be329 · f0be329
1 parent 32c769f
commit f0be329
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cirq/contrib/routing/device.py b/cirq/contrib/routing/device.py
@@ -59,6 +59,9 @@ def nx_qubit_layout(graph: nx.Graph) \
 
     >>> import cirq.contrib.routing as ccr
     >>> import networkx as nx
+    >>> import matplotlib.pyplot as plt
+    >>> # Clear plot state to prevent issues with pyplot dimensionality.
+    >>> plt.clf()
     >>> g = ccr.xmon_device_to_graph(cirq.google.Foxtail)
     >>> pos = ccr.nx_qubit_layout(g)
     >>> nx.draw_networkx(g, pos=pos)