From e43b13e2c2892f7497e8dcf5bef9b81023a01ea4 Mon Sep 17 00:00:00 2001
From: Lourens Veen <l.veen@esciencecenter.nl>
Date: Wed, 17 Jul 2024 18:13:16 +0200
Subject: [PATCH] Fix mypy type error

---
 muscle3/profiling.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/muscle3/profiling.py b/muscle3/profiling.py
index b705a8c5..a7ac43bb 100644
--- a/muscle3/profiling.py
+++ b/muscle3/profiling.py
@@ -1,9 +1,10 @@
 import sqlite3
 from pathlib import Path
-from typing import List, Optional, Tuple
+from typing import cast, List, Optional, Tuple
 
 import numpy as np
 from matplotlib.axes import Axes
+from matplotlib.figure import Figure
 from matplotlib import pyplot as plt
 from matplotlib.patches import Rectangle
 
@@ -237,7 +238,7 @@ def __init__(self, performance_file: Path) -> None:
         ordered_names.insert(6, 'RUNNING')
 
         ax.legend(ordered_artists, ordered_names, loc='upper right')
-        ax.figure.canvas.draw_idle()
+        cast(Figure, ax.figure).canvas.draw_idle()
 
     def close(self) -> None:
         """Closes the database connection"""