From c203a4f735f5a243b37f918674bd8b8f9558f2ff Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 23 Jan 2024 23:24:26 +0100
Subject: [PATCH] [scripts.train_analysis] Fix typing
---
src/mednet/scripts/train_analysis.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/mednet/scripts/train_analysis.py b/src/mednet/scripts/train_analysis.py
index 94867b6a..7736d2cb 100644
--- a/src/mednet/scripts/train_analysis.py
+++ b/src/mednet/scripts/train_analysis.py
@@ -66,6 +66,8 @@ def create_figures(
import matplotlib.pyplot as plt
+ from matplotlib.axes import Axes
+ from matplotlib.figure import Figure
from matplotlib.ticker import MaxNLocator
figures = []
@@ -77,8 +79,8 @@ def create_figures(
continue
fig, ax = plt.subplots(1, 1)
- ax = typing.cast(plt.Axes, ax)
- fig = typing.cast(plt.Figure, fig)
+ ax = typing.cast(Axes, ax)
+ fig = typing.cast(Figure, fig)
if len(curves) == 1:
# there is only one curve, just plot it
--
GitLab