fix: changed import of matplotlib.pyplot

qiboteam · Dec 5, 2024 · cd3fa6a · cd3fa6a
1 parent a518e99
commit cd3fa6a
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"],