From 44901bdcd1df8340541689a935ebccfa3251951e Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 12 Jun 2024 17:57:27 +0200 Subject: [PATCH] Fix deprecated-typing-alias pylint warnings in zha tests (#119453) --- tests/components/zha/test_registries.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/components/zha/test_registries.py b/tests/components/zha/test_registries.py index 18253186cf1..2b1c0dcc561 100644 --- a/tests/components/zha/test_registries.py +++ b/tests/components/zha/test_registries.py @@ -2,20 +2,18 @@ from __future__ import annotations -import typing from unittest import mock import pytest +from typing_extensions import Generator import zigpy.quirks as zigpy_quirks from homeassistant.components.zha.binary_sensor import IASZone from homeassistant.components.zha.core import registries from homeassistant.components.zha.core.const import ATTR_QUIRK_ID +from homeassistant.components.zha.entity import ZhaEntity from homeassistant.helpers import entity_registry as er -if typing.TYPE_CHECKING: - from homeassistant.components.zha.core.entity import ZhaEntity - MANUFACTURER = "mock manufacturer" MODEL = "mock model" QUIRK_CLASS = "mock.test.quirk.class" @@ -532,7 +530,7 @@ def test_multi_sensor_match( } -def iter_all_rules() -> typing.Iterable[registries.MatchRule, list[type[ZhaEntity]]]: +def iter_all_rules() -> Generator[tuple[registries.MatchRule, list[type[ZhaEntity]]]]: """Iterate over all match rules and their corresponding entities.""" for rules in registries.ZHA_ENTITIES._strict_registry.values():