update_forward_refs() calling to prevent field "trigger" not yet prep…

…ared error
Leggin · May 14, 2024 · 2d3b8bd · 2d3b8bd
1 parent 23709ff
commit 2d3b8bd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/dirigera/devices/scene.py b/src/dirigera/devices/scene.py
@@ -129,6 +129,9 @@ def trigger(self) -> None:
     def undo(self) -> None:
         self.dirigera_client.post(route=f"/scenes/{self.id}/undo")
 
+Trigger.update_forward_refs()
+Action.update_forward_refs()
+Scene.update_forward_refs()
 
 def dict_to_scene(data: Dict[str, Any], dirigera_client: AbstractSmartHomeHub) -> Scene:
     return Scene(dirigeraClient=dirigera_client, **data)