Merge branch 'master' into Heisenberg

qiboteam · Dec 8, 2024 · fa8c6f9 · fa8c6f9
2 parents adb63b6 + 9e00ebd
commit fa8c6f9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/qibo/ui/mpldrawer.py b/src/qibo/ui/mpldrawer.py
@@ -7,6 +7,7 @@
 from typing import Union
 
 import matplotlib
+import matplotlib.pyplot as plt
 import numpy as np
 
 from qibo import gates
@@ -377,7 +378,7 @@ def _swapx(ax, x, y, plot_params):
 
 def _setup_figure(nq, ng, gate_grid, wire_grid, plot_params):
     scale = plot_params["scale"]
-    fig = matplotlib.pyplot.figure(
+    fig = plt.figure(
         figsize=(ng * scale, nq * scale),
         facecolor=plot_params["facecolor"],
         edgecolor=plot_params["edgecolor"],