diff --git a/.strict-typing b/.strict-typing index 0db793ace7a..6eeb39ceddb 100644 --- a/.strict-typing +++ b/.strict-typing @@ -153,6 +153,7 @@ homeassistant.components.elgato.* homeassistant.components.elkm1.* homeassistant.components.emulated_hue.* homeassistant.components.energy.* +homeassistant.components.energyzero.* homeassistant.components.enigma2.* homeassistant.components.esphome.* homeassistant.components.event.* diff --git a/homeassistant/components/energyzero/services.py b/homeassistant/components/energyzero/services.py index d8e548c22f8..325c443375e 100644 --- a/homeassistant/components/energyzero/services.py +++ b/homeassistant/components/energyzero/services.py @@ -109,7 +109,8 @@ def __get_coordinator( }, ) - return hass.data[DOMAIN][entry_id] + coordinator: EnergyZeroDataUpdateCoordinator = hass.data[DOMAIN][entry_id] + return coordinator async def __get_prices( diff --git a/mypy.ini b/mypy.ini index 30b4ee9f048..fc020a7b209 100644 --- a/mypy.ini +++ b/mypy.ini @@ -1291,6 +1291,16 @@ disallow_untyped_defs = true warn_return_any = true warn_unreachable = true +[mypy-homeassistant.components.energyzero.*] +check_untyped_defs = true +disallow_incomplete_defs = true +disallow_subclassing_any = true +disallow_untyped_calls = true +disallow_untyped_decorators = true +disallow_untyped_defs = true +warn_return_any = true +warn_unreachable = true + [mypy-homeassistant.components.enigma2.*] check_untyped_defs = true disallow_incomplete_defs = true