From 86d5f6bbc0a25cd3e1f456f070cbdc9d6ad4626a Mon Sep 17 00:00:00 2001
From: Normann <github@koldrack.com>
Date: Mon, 13 Jan 2025 19:19:33 +0100
Subject: [PATCH] mypy fixes

---
 src/akkudoktoreos/utils/visualize.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/akkudoktoreos/utils/visualize.py b/src/akkudoktoreos/utils/visualize.py
index 2ac0e369..4850b8f7 100644
--- a/src/akkudoktoreos/utils/visualize.py
+++ b/src/akkudoktoreos/utils/visualize.py
@@ -111,7 +111,7 @@ def _save_group_to_pdf(self, group: list[Callable[[], None]]) -> None:
     def create_line_chart_date(
         self,
         start_date: pendulum.DateTime,
-        y_list: list[list[Optional[float]]],
+        y_list: list[Union[np.ndarray, list[Optional[float]], list[float]]],
         ylabel: str,
         xlabel: Optional[str] = None,
         title: Optional[str] = None,
@@ -149,8 +149,8 @@ def chart() -> None:
             # Auto-format the x-axis for readability
 
             # Move major tick labels further down to avoid collision with minor tick labels
-            for label in plt.gca().get_xticklabels(which="major"):
-                label.set_y(-0.04)
+            for plt_label in plt.gca().get_xticklabels(which="major"):
+                plt_label.set_y(-0.04)
 
             # Add labels, title, and legend
             if xlabel:
@@ -414,7 +414,9 @@ def prepare_visualize(
     print(parameters.ems.gesamtlast)
     report.create_line_chart_date(
         next_full_hour_date,  # start_date
-        [parameters.ems.gesamtlast],
+        [
+            parameters.ems.gesamtlast,
+        ],
         title="Load Profile",
         # xlabel="Hours", # not enough space
         ylabel="Load (Wh)",
@@ -422,7 +424,9 @@ def prepare_visualize(
     )
     report.create_line_chart_date(
         next_full_hour_date,  # start_date
-        [parameters.ems.pv_prognose_wh],
+        [
+            parameters.ems.pv_prognose_wh,
+        ],
         title="PV Forecast",
         # xlabel="Hours", # not enough space
         ylabel="PV Generation (Wh)",
@@ -439,7 +443,9 @@ def prepare_visualize(
     if parameters.temperature_forecast:
         report.create_line_chart_date(
             next_full_hour_date,  # start_date
-            [parameters.temperature_forecast],
+            [
+                parameters.temperature_forecast,
+            ],
             title="Temperature Forecast",
             # xlabel="Hours", # not enough space
             ylabel="°C",