diff --git a/homeassistant/components/gtfs/sensor.py b/homeassistant/components/gtfs/sensor.py index f8f89b1ea36..f97bc9796ec 100644 --- a/homeassistant/components/gtfs/sensor.py +++ b/homeassistant/components/gtfs/sensor.py @@ -11,7 +11,10 @@ import pygtfs from sqlalchemy.sql import text import voluptuous as vol -from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity +from homeassistant.components.sensor import ( + PLATFORM_SCHEMA as SENSOR_PLATFORM_SCHEMA, + SensorEntity, +) from homeassistant.const import ( ATTR_ATTRIBUTION, CONF_NAME, @@ -254,8 +257,8 @@ WHEELCHAIR_ACCESS_OPTIONS = {1: True, 2: False} WHEELCHAIR_BOARDING_DEFAULT = STATE_UNKNOWN WHEELCHAIR_BOARDING_OPTIONS = {1: True, 2: False} -PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( - { # type: ignore +PLATFORM_SCHEMA = SENSOR_PLATFORM_SCHEMA.extend( + { vol.Required(CONF_ORIGIN): cv.string, vol.Required(CONF_DESTINATION): cv.string, vol.Required(CONF_DATA): cv.string, @@ -490,7 +493,7 @@ def setup_platform( origin = config.get(CONF_ORIGIN) destination = config.get(CONF_DESTINATION) name = config.get(CONF_NAME) - offset = config.get(CONF_OFFSET) + offset: datetime.timedelta = config[CONF_OFFSET] include_tomorrow = config[CONF_TOMORROW] if not os.path.exists(gtfs_dir): @@ -541,10 +544,10 @@ class GTFSDepartureSensor(SensorEntity): self._icon = ICON self._name = "" self._state: str | None = None - self._attributes = {} + self._attributes: dict[str, Any] = {} self._agency = None - self._departure = {} + self._departure: dict[str, Any] = {} self._destination = None self._origin = None self._route = None @@ -559,7 +562,7 @@ class GTFSDepartureSensor(SensorEntity): return self._name @property - def native_value(self) -> str | None: # type: ignore + def native_value(self) -> str | None: """Return the state of the sensor.""" return self._state diff --git a/mypy.ini b/mypy.ini index e566b7f1898..82ed7d6ae9d 100644 --- a/mypy.ini +++ b/mypy.ini @@ -1355,9 +1355,6 @@ ignore_errors = true [mypy-homeassistant.components.growatt_server.*] ignore_errors = true -[mypy-homeassistant.components.gtfs.*] -ignore_errors = true - [mypy-homeassistant.components.habitica.*] ignore_errors = true diff --git a/script/hassfest/mypy_config.py b/script/hassfest/mypy_config.py index bb2cf72b72e..581b4865f7c 100644 --- a/script/hassfest/mypy_config.py +++ b/script/hassfest/mypy_config.py @@ -44,7 +44,6 @@ IGNORED_MODULES: Final[list[str]] = [ "homeassistant.components.gpmdp.*", "homeassistant.components.gree.*", "homeassistant.components.growatt_server.*", - "homeassistant.components.gtfs.*", "homeassistant.components.habitica.*", "homeassistant.components.harmony.*", "homeassistant.components.hassio.*",