From a081a46caa0e33b8e4171955cd01c67630ec15ac Mon Sep 17 00:00:00 2001
From: Teemu Rytilahti <tpr@iki.fi>
Date: Sat, 30 Nov 2024 16:06:00 +0100
Subject: [PATCH] Cleanup a bit

---
 kasa/smart/modules/vacuum.py | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/kasa/smart/modules/vacuum.py b/kasa/smart/modules/vacuum.py
index e84b0ab2..6e2955e7 100644
--- a/kasa/smart/modules/vacuum.py
+++ b/kasa/smart/modules/vacuum.py
@@ -1,4 +1,4 @@
-"""Implementation of vacuum (experimental)."""
+"""Implementation of vacuum."""
 
 from __future__ import annotations
 
@@ -30,11 +30,11 @@ class Status(IntEnum):
 
 
 class Vacuum(SmartModule):
-    """Implementation of experimental vacuum support."""
+    """Implementation of vacuum support."""
 
     REQUIRED_COMPONENT = "clean"
 
-    def __init__(self, device: SmartDevice, module: str):
+    def __init__(self, device: SmartDevice, module: str) -> None:
         super().__init__(device, module)
         self._add_feature(
             Feature(
@@ -85,7 +85,7 @@ class Vacuum(SmartModule):
         """Query to execute during the update cycle."""
         return {"getVacStatus": None}
 
-    async def start(self) -> None:
+    async def start(self) -> dict:
         """Start cleaning."""
         # If we are paused, do not restart cleaning
 
@@ -103,23 +103,23 @@ class Vacuum(SmartModule):
             },
         )
 
-    async def pause(self):
+    async def pause(self) -> dict:
         """Pause cleaning."""
         return await self.set_pause(True)
 
-    async def resume(self):
+    async def resume(self) -> dict:
         """Resume cleaning."""
         return await self.set_pause(False)
 
-    async def set_pause(self, enabled: bool) -> None:
+    async def set_pause(self, enabled: bool) -> dict:
         """Pause or resume cleaning."""
         return await self.call("setRobotPause", {"pause": enabled})
 
-    async def return_home(self):
+    async def return_home(self) -> dict:
         """Return home."""
         return await self.set_return_home(True)
 
-    async def set_return_home(self, enabled: bool) -> None:
+    async def set_return_home(self, enabled: bool) -> dict:
         """Return home / pause returning."""
         return await self.call("setSwitchCharge", {"switch_charge": enabled})
 
@@ -128,9 +128,10 @@ class Vacuum(SmartModule):
         """Return current status."""
         if self.data.get("err_status"):
             return Status.Error
+
         status_code = self.data["status"]
         try:
             return Status(status_code)
-        except Exception:
+        except ValueError:
             _LOGGER.warning("Got unknown status code: %s (%s)", status_code, self.data)
             return Status.Unknown