Clean up

Author: bouwew

plugwise/__init__.py

@@ -461,7 +461,9 @@ async def set_regulation_mode(self, mode: str) -> None:
                 f"Failed to set regulation mode: {str(exc)}"
             ) from exc  # pragma no cover
 
-    async def set_dhw_mode(self, key: str, location: str, length: int, mode: str) -> None:
+    async def set_dhw_mode(
+        self, key: str, location: str, length: int, mode: str
+    ) -> None:
         """Set the domestic hot water heating regulation mode."""
         try:  # pragma no cover
             await self._smile_api.set_dhw_mode(

plugwise/legacy/smile.py

@@ -131,7 +131,9 @@ async def delete_notification(self) -> None:
     async def reboot_gateway(self) -> None:
         """Set-function placeholder for legacy devices."""
 
-    async def set_dhw_mode(self, key: str, location: str, length: int, mode: str) -> None:
+    async def set_dhw_mode(
+        self, key: str, location: str, length: int, mode: str
+    ) -> None:
         """Set-function placeholder for legacy devices."""
 
     async def set_gateway_mode(self, mode: str) -> None:

plugwise/smile.py

@@ -16,7 +16,6 @@
     DOMAIN_OBJECTS,
     GATEWAY_REBOOT,
     LOCATIONS,
-    LOGGER,
     MAX_SETPOINT,
     MIN_SETPOINT,
     NONE,
@@ -250,9 +249,7 @@ async def set_select(
             case "select_zone_profile":
                 await self.set_zone_profile(loc_id, option)
 
-    async def set_dhw_mode(
-        self, key: str, loc_id: str, length: int, mode: str
-    ) -> None:
+    async def set_dhw_mode(self, key: str, loc_id: str, length: int, mode: str) -> None:
         """Set the domestic hot water mode."""
         if mode not in self._dhw_allowed_modes:
             raise PlugwiseError("Plugwise: invalid dhw mode.")