Merge pull request #45 from slajob/main

update Trigger field reference to prevent pydantic validation error
sanjoyg · May 14, 2024 · a66bbce · a66bbce
2 parents a96979c + 1a4ec7f
commit a66bbce
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/custom_components/dirigera_platform/scene.py b/custom_components/dirigera_platform/scene.py
@@ -5,6 +5,7 @@
 
 from dirigera import Hub
 from dirigera.devices.scene import Scene as DirigeraScene
+from dirigera.devices.scene import Trigger, TriggerDetails
 
 from homeassistant.components.scene import Scene
 from homeassistant.config_entries import ConfigEntry
@@ -26,6 +27,9 @@ async def async_setup_entry(
     config = hass.data[DOMAIN][entry.entry_id]
     hub = Hub(config[CONF_TOKEN], config[CONF_IP_ADDRESS])
 
+    Trigger.update_forward_refs()
+    TriggerDetails.update_forward_refs()
+
     # TODO: Add mock scenes
     if config[CONF_IP_ADDRESS] == "mock":
         return