mypy

starkillerOG · Feb 27, 2025 · 5672bf2 · 5672bf2
1 parent 2231839
commit 5672bf2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/reolink_aio/baichuan/baichuan.py b/reolink_aio/baichuan/baichuan.py
@@ -817,12 +817,13 @@ def api_version(self, capability: str, channel: int | None = None, no_key_return
     @property
     def abilities(self) -> dict[int | str, Any]:
         """Return the abilities as a dictionary"""
-        abilities_dict = {}
+        abilities_dict: dict[int | str, dict[str, int | str]] = {}
         for key, xml in self._abilities.items():
-            pretty_key = key if key is not None else "Host"
+            pretty_key: str | int = key if key is not None else "Host"
             abilities_dict[pretty_key] = {}
             for feature in xml:
                 if feature.text is not None:
+                    value: int | str
                     try:
                         value = int(feature.text)
                     except ValueError: