fix imgui_fig

pthom · Oct 6, 2024 · 92d2d44 · 92d2d44
1 parent 1e2166e
commit 92d2d44
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindings/imgui_bundle/imgui_fig.py b/bindings/imgui_bundle/imgui_fig.py
@@ -7,6 +7,7 @@
 """
 import matplotlib
 matplotlib.use('Agg')  # set the renderer to Tk
+from matplotlib.figure import Figure  # noqa: E402
 import numpy  # noqa: E402
 import cv2  # noqa: E402
 import matplotlib  # noqa: E402
@@ -15,7 +16,7 @@
 
 
 @static(fig_image_cache=dict())
-def _fig_to_image(label_id: str, figure: matplotlib.figure.Figure, refresh_image: bool = False) -> numpy.ndarray:
+def _fig_to_image(label_id: str, figure: Figure, refresh_image: bool = False) -> numpy.ndarray:
     """
     Convert a Matplotlib figure to an RGB image.