Fixing types

FastyBird · Feb 1, 2022 · 10d7ada · 10d7ada
1 parent f57f899
commit 10d7ada
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastybird_fb_bus_connector/registry/model.py b/fastybird_fb_bus_connector/registry/model.py
@@ -800,7 +800,7 @@ def reset(self, device_id: Optional[uuid.UUID] = None, registers_type: Optional[
     def set_actual_value(
         self,
         register: RegisterRecord,
-        value: Union[str, int, float, bool, datetime, ButtonPayload, SwitchPayload],
+        value: Union[str, int, float, bool, datetime, ButtonPayload, SwitchPayload, None],
     ) -> RegisterRecord:
         """Set actual value to register"""
         existing_record = self.get_by_id(register_id=register.id)

diff --git a/fastybird_fb_bus_connector/registry/records.py b/fastybird_fb_bus_connector/registry/records.py
@@ -433,7 +433,7 @@ def actual_value(self) -> Union[str, int, float, bool, datetime, ButtonPayload,
     # -----------------------------------------------------------------------------
 
     @actual_value.setter
-    def actual_value(self, value: Union[str, int, float, bool, datetime, ButtonPayload, SwitchPayload]) -> None:
+    def actual_value(self, value: Union[str, int, float, bool, datetime, ButtonPayload, SwitchPayload, None]) -> None:
         """Set register actual value"""
         self._actual_value = value