Allow empty capabilities (#293)

Fixes #281
pschmitt · Apr 1, 2024 · a6da692 · a6da692
1 parent 0d058c4
commit a6da692
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/roombapy/roomba_info.py b/roombapy/roomba_info.py
@@ -8,6 +8,8 @@
 from mashumaro import field_options
 from mashumaro.mixins.orjson import DataClassORJSONMixin
 
+Capabilities = dict[str, int] | None
+
 
 @dataclass
 class RoombaInfo(DataClassORJSONMixin):
@@ -19,7 +21,7 @@ class RoombaInfo(DataClassORJSONMixin):
     mac: str
     robot_name: str = field(metadata=field_options(alias="robotname"))
     sku: str
-    capabilities: dict[str, int] = field(metadata=field_options(alias="cap"))
+    capabilities: Capabilities = field(metadata=field_options(alias="cap"))
     password: str | None = None
 
     @cached_property