@@ -77,8 +77,7 @@ class CleaningMode(StrEnum):
7777 Prefer this abstraction when you want to present or switch between the
7878 user-facing cleaning behaviors exposed by the app. The lower-level
7979 `VacuumModes`, `WaterModes`, and `CleanRoutes` enums are still useful for
80- advanced tuning, but most integrations should treat them as implementation
81- details of a single high-level cleaning mode.
80+ fine-grained selection.
8281 """
8382
8483 VACUUM = "vacuum"
@@ -330,9 +329,9 @@ def get_current_cleaning_mode(
330329
331330
332331def is_mode_customized (
333- clean_mode : int | VacuumModes | None ,
334- water_mode : int | WaterModes | None ,
335- mop_mode : int | CleanRoutes | None ,
332+ clean_mode : VacuumModes | None ,
333+ water_mode : WaterModes | None ,
334+ mop_mode : CleanRoutes | None ,
336335) -> bool :
337336 """Check if any of the cleaning modes are set to a custom value."""
338337 return (
@@ -343,9 +342,9 @@ def is_mode_customized(
343342
344343
345344def is_smart_mode_set (
346- water_mode : int | WaterModes | None ,
347- clean_mode : int | VacuumModes | None ,
348- mop_mode : int | CleanRoutes | None ,
345+ water_mode : WaterModes | None ,
346+ clean_mode : VacuumModes | None ,
347+ mop_mode : CleanRoutes | None ,
349348) -> bool :
350349 """Check if the smart mode is set for the given water mode and clean mode"""
351350 return (
0 commit comments