From 8e6fa54e0a297a8d933747d3025647f1006bce0f Mon Sep 17 00:00:00 2001 From: IceBotYT <34712694+IceBotYT@users.noreply.github.com> Date: Tue, 14 Jun 2022 15:08:47 -0400 Subject: [PATCH] Improve PECO integration (#73460) --- homeassistant/components/peco/__init__.py | 3 --- homeassistant/components/peco/sensor.py | 4 +--- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/homeassistant/components/peco/__init__.py b/homeassistant/components/peco/__init__.py index 6f88bf36c50..86f69213a1c 100644 --- a/homeassistant/components/peco/__init__.py +++ b/homeassistant/components/peco/__init__.py @@ -1,7 +1,6 @@ """The PECO Outage Counter integration.""" from __future__ import annotations -import asyncio from dataclasses import dataclass from datetime import timedelta from typing import Final @@ -48,8 +47,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: raise UpdateFailed(f"Error fetching data: {err}") from err except BadJSONError as err: raise UpdateFailed(f"Error parsing data: {err}") from err - except asyncio.TimeoutError as err: - raise UpdateFailed(f"Timeout fetching data: {err}") from err return data coordinator = DataUpdateCoordinator( diff --git a/homeassistant/components/peco/sensor.py b/homeassistant/components/peco/sensor.py index dfd354f5c03..5afc300bfa8 100644 --- a/homeassistant/components/peco/sensor.py +++ b/homeassistant/components/peco/sensor.py @@ -93,10 +93,8 @@ async def async_setup_entry( coordinator = hass.data[DOMAIN][config_entry.entry_id] async_add_entities( - [PecoSensor(sensor, county, coordinator) for sensor in SENSOR_LIST], - True, + PecoSensor(sensor, county, coordinator) for sensor in SENSOR_LIST ) - return class PecoSensor(