Update __init__.py

azerty9971 · Jan 26, 2025 · 051c7e1 · 051c7e1
1 parent 86073d4
commit 051c7e1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/xtend_tuya/__init__.py b/custom_components/xtend_tuya/__init__.py
@@ -1,9 +1,9 @@
 """Support for Tuya Smart devices."""
 
-from __future__ import absolute_import
+from __future__ import annotations
 import logging
 
-import time
+import time as py_time
 
 from homeassistant.config_entries import ConfigEntry, ConfigEntryState
 from homeassistant.core import HomeAssistant
@@ -90,10 +90,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: XTConfigEntry) -> bool:
 async def cleanup_device_registry(hass: HomeAssistant, multi_manager: MultiManager, current_entry: ConfigEntry) -> None:
     """Remove deleted device registry entry if there are no remaining entities."""
     while not are_all_domain_config_loaded(hass, DOMAIN_ORIG, None):
-        time.sleep(1)
+        py_time.sleep(1)
     while not are_all_domain_config_loaded(hass, DOMAIN, current_entry):
         if is_config_entry_master(hass, DOMAIN, current_entry):
-            time.sleep(1)
+            py_time.sleep(1)
         else:
             return
     device_registry = dr.async_get(hass)