mypy tweaks

zxdavb · Sep 1, 2024 · 9bc985e · 9bc985e
1 parent d320c47
commit 9bc985e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/src/evohomeasync2/system.py b/src/evohomeasync2/system.py
@@ -133,7 +133,7 @@ def __init__(self, gateway: Gateway, config: _EvoDictT, /) -> None:
         self._status: _EvoDictT = {}
 
         self.zones: list[Zone] = []
-        self.zone_by_id: dict[str, Zone] = {}
+        self.zone_by_id: dict[str, Zone | HotWater] = {}
         self.hotwater: None | HotWater = None
 
         zon_config: _EvoDictT

diff --git a/tests/tests_rf/test_v2_apis.py b/tests/tests_rf/test_v2_apis.py
@@ -4,7 +4,6 @@
 from __future__ import annotations
 
 from datetime import datetime as dt
-from typing import TYPE_CHECKING
 
 import pytest
 
@@ -26,9 +25,6 @@
 from .const import ExitTestReason
 from .helpers import instantiate_client_v2
 
-if TYPE_CHECKING:
-    from evohomeasync2.zone import Zone
-
 #######################################################################################
 
 
@@ -115,16 +111,14 @@ async def _test_sched__apis(evo: evo2.EvohomeClient) -> None:
         assert SCH_PUT_SCHEDULE_DHW(schedule)
         await dhw.set_schedule(schedule)
 
-    zone: Zone | None
-
     if (zone := evo.default_system().zones[0]) and zone._id != faked.GHOST_ZONE_ID:
         schedule = await zone.get_schedule()
         assert SCH_PUT_SCHEDULE_ZONE(schedule)
         await zone.set_schedule(schedule)
 
-    if zone := evo.default_system().zone_by_id.get(faked.GHOST_ZONE_ID):
+    if child := evo.default_system().zone_by_id.get(faked.GHOST_ZONE_ID):
         try:
-            schedule = await zone.get_schedule()
+            schedule = await child.get_schedule()
         except evo2.InvalidScheduleError:
             pass
         else: