Update entity.py

RenierM26 · Jan 8, 2023 · 40a3fe3 · 40a3fe3
1 parent ecf8193
commit 40a3fe3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/ezviz_cloud/entity.py b/custom_components/ezviz_cloud/entity.py
@@ -1,9 +1,9 @@
-"""An abstract class common to all Ezviz entities."""
+"""An abstract class common to all EZVIZ entities."""
 from __future__ import annotations
 
 from typing import Any
 
-from homeassistant.helpers import device_registry
+from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
 from homeassistant.helpers.entity import DeviceInfo, Entity
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
@@ -12,7 +12,7 @@
 
 
 class EzvizEntity(CoordinatorEntity[EzvizDataUpdateCoordinator], Entity):
-    """Generic entity encapsulating common features of Ezviz device."""
+    """Generic entity encapsulating common features of EZVIZ device."""
 
     def __init__(
         self,
@@ -26,7 +26,7 @@ def __init__(
         self._attr_device_info = DeviceInfo(
             identifiers={(DOMAIN, serial)},
             connections={
-                (device_registry.CONNECTION_NETWORK_MAC, self.data["mac_address"]),
+                (CONNECTION_NETWORK_MAC, self.data["mac_address"]),
             },
             manufacturer=MANUFACTURER,
             model=self.data["device_sub_category"],