diff --git a/custom_components/home_assistant_covid19_augsburg/__init__.py b/custom_components/home_assistant_covid19_augsburg/__init__.py index bb9fe06..a331f13 100644 --- a/custom_components/home_assistant_covid19_augsburg/__init__.py +++ b/custom_components/home_assistant_covid19_augsburg/__init__.py @@ -12,7 +12,7 @@ if TYPE_CHECKING: from homeassistant.core import HomeAssistant from .const import DOMAIN -from .crawler import CovidCrawler, IncidenceData +from .crawler import CovidCrawler _LOGGER = logging.getLogger(__name__) diff --git a/custom_components/home_assistant_covid19_augsburg/sensor.py b/custom_components/home_assistant_covid19_augsburg/sensor.py index 558022b..2307bb7 100644 --- a/custom_components/home_assistant_covid19_augsburg/sensor.py +++ b/custom_components/home_assistant_covid19_augsburg/sensor.py @@ -9,7 +9,12 @@ async def async_setup_entry(hass, _, async_add_entities): """Defer sensor setup to the shared sensor module.""" coordinator = await get_coordinator(hass) - async_add_entities([CoronaAugsburgSensor(coordinator)]) + async_add_entities( + [ + CoronaAugsburgSensor(coordinator), + CoronaAugsburgVaccinationSensor(coordinator), + ] + ) class CoronaAugsburgSensor(Entity):