diff --git a/.strict-typing b/.strict-typing index f81b6249fed..77f7b6f50c2 100644 --- a/.strict-typing +++ b/.strict-typing @@ -226,6 +226,7 @@ homeassistant.components.tplink.* homeassistant.components.tolo.* homeassistant.components.tractive.* homeassistant.components.tradfri.* +homeassistant.components.trafikverket_ferry.* homeassistant.components.trafikverket_train.* homeassistant.components.trafikverket_weatherstation.* homeassistant.components.tts.* diff --git a/homeassistant/components/trafikverket_ferry/sensor.py b/homeassistant/components/trafikverket_ferry/sensor.py index bab73d72210..256341a7132 100644 --- a/homeassistant/components/trafikverket_ferry/sensor.py +++ b/homeassistant/components/trafikverket_ferry/sensor.py @@ -4,7 +4,7 @@ from __future__ import annotations from collections.abc import Callable from dataclasses import dataclass from datetime import datetime, timedelta -from typing import Any +from typing import Any, cast from homeassistant.components.sensor import ( SensorDeviceClass, @@ -55,21 +55,21 @@ SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = ( icon="mdi:clock", device_class=SensorDeviceClass.TIMESTAMP, value_fn=lambda data: as_utc(data["departure_time"]), - info_fn=lambda data: data["departure_information"], + info_fn=lambda data: cast(list[str], data["departure_information"]), ), TrafikverketSensorEntityDescription( key="departure_from", name="Departure From", icon="mdi:ferry", - value_fn=lambda data: data["departure_from"], - info_fn=lambda data: data["departure_information"], + value_fn=lambda data: cast(str, data["departure_from"]), + info_fn=lambda data: cast(list[str], data["departure_information"]), ), TrafikverketSensorEntityDescription( key="departure_to", name="Departure To", icon="mdi:ferry", - value_fn=lambda data: data["departure_to"], - info_fn=lambda data: data["departure_information"], + value_fn=lambda data: cast(str, data["departure_to"]), + info_fn=lambda data: cast(list[str], data["departure_information"]), ), TrafikverketSensorEntityDescription( key="departure_modified", @@ -77,7 +77,7 @@ SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = ( icon="mdi:clock", device_class=SensorDeviceClass.TIMESTAMP, value_fn=lambda data: as_utc(data["departure_modified"]), - info_fn=lambda data: data["departure_information"], + info_fn=lambda data: cast(list[str], data["departure_information"]), entity_registry_enabled_default=False, ), TrafikverketSensorEntityDescription( diff --git a/mypy.ini b/mypy.ini index 11a2d83ec40..9e27addae89 100644 --- a/mypy.ini +++ b/mypy.ini @@ -2250,6 +2250,17 @@ no_implicit_optional = true warn_return_any = true warn_unreachable = true +[mypy-homeassistant.components.trafikverket_ferry.*] +check_untyped_defs = true +disallow_incomplete_defs = true +disallow_subclassing_any = true +disallow_untyped_calls = true +disallow_untyped_decorators = true +disallow_untyped_defs = true +no_implicit_optional = true +warn_return_any = true +warn_unreachable = true + [mypy-homeassistant.components.trafikverket_train.*] check_untyped_defs = true disallow_incomplete_defs = true