Fix

Author: bouwew

plugwise/__init__.py

@@ -76,7 +76,7 @@ def __init__(
         self._target_smile: str = NONE
         self.gateway_id: str = NONE
         self.loc_data: dict[str, ThermoLoc] = {}
-        self.smile_fw_version: Version | None
+        self.smile_fw_version: Version | None = None
         self.smile_hostname: str = NONE
         self.smile_hw_version: str | None = None
         self.smile_legacy = False
@@ -88,7 +88,7 @@ def __init__(
         self.smile_version: Version | None = None
         self.smile_zigbee_mac_address: str | None = None
 
-    async def connect(self) -> str | None:
+    async def connect(self) -> Version | None:
         """Connect to Plugwise device and determine its name, type and version."""
         result = await self._request(DOMAIN_OBJECTS)
         # Work-around for Stretch fw 2.7.18

plugwise/legacy/smile.py

@@ -29,6 +29,7 @@
 
 import aiohttp
 from munch import Munch
+from packaging.version import Version
 
 
 class SmileLegacyAPI(SmileLegacyData):
@@ -48,7 +49,7 @@ def __init__(
         _stretch_v2: bool,
         _target_smile: str,
         loc_data: dict[str, ThermoLoc],
-        smile_fw_version: str | None,
+        smile_fw_version: Version | None,
         smile_hostname: str,
         smile_hw_version: str | None,
         smile_mac_address: str | None,

plugwise/smile.py

@@ -35,6 +35,7 @@
 
 # Dict as class
 from munch import Munch
+from packaging.version import Version
 
 
 class SmileAPI(SmileData):
@@ -57,7 +58,7 @@ def __init__(
         _schedule_old_states: dict[str, dict[str, str]],
         gateway_id: str,
         loc_data: dict[str, ThermoLoc],
-        smile_fw_version: str | None,
+        smile_fw_version: Version | None,
         smile_hostname: str | None,
         smile_hw_version: str | None,
         smile_mac_address: str | None,

tests/test_init.py

@@ -2,7 +2,6 @@
 """Test Plugwise Home Assistant module and generate test JSON fixtures."""
 import importlib
 import json
-from packaging import version
 
 # Fixture writing
 import logging
@@ -18,6 +17,7 @@
 # Testing
 import aiohttp
 from freezegun import freeze_time
+from packaging import version
 
 pw_constants = importlib.import_module("plugwise.constants")
 pw_exceptions = importlib.import_module("plugwise.exceptions")