Merge pull request #224 from Snuffy2/Optimize-Device-Trackers

Optimize Device Trackers
travisghansen · Sep 30, 2024 · 5f5e407 · 5f5e407
2 parents b78e85a + c7d5639
commit 5f5e407
Show file tree

Hide file tree

Showing 4 changed files with 244 additions and 240 deletions.
diff --git a/custom_components/opnsense/__init__.py b/custom_components/opnsense/__init__.py
@@ -20,7 +20,6 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.entity_registry import async_get
 from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
-from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.typing import ConfigType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.util import slugify
@@ -259,7 +258,7 @@ async def async_remove_entity(self, entity):
             registry.async_remove(entity.entity_id)
 
 
-class OPNsenseEntity(CoordinatorEntity[OPNsenseDataUpdateCoordinator], RestoreEntity):
+class OPNsenseEntity(CoordinatorEntity[OPNsenseDataUpdateCoordinator]):
     """Base entity for OPNsense"""
 
     def __init__(