Activate mypy. (home-assistant#55965)

katiebot · Sep 18, 2021 · 476d04e · 476d04e
1 parent 39bc127
commit 476d04e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/homeassistant/components/entur_public_transport/sensor.py b/homeassistant/components/entur_public_transport/sensor.py
@@ -1,4 +1,6 @@
 """Real-time information about public transport departures in Norway."""
+from __future__ import annotations
+
 from datetime import datetime, timedelta
 
 from enturclient import EnturPublicTransportData
@@ -158,17 +160,17 @@ def __init__(
         self._stop = stop
         self._show_on_map = show_on_map
         self._name = name
-        self._state = None
+        self._state: int | None = None
         self._icon = ICONS[DEFAULT_ICON_KEY]
-        self._attributes = {}
+        self._attributes: dict[str, str] = {}
 
     @property
     def name(self) -> str:
         """Return the name of the sensor."""
         return self._name
 
     @property
-    def native_value(self) -> str:
+    def native_value(self) -> int | None:
         """Return the state of the sensor."""
         return self._state
 
@@ -195,7 +197,7 @@ async def async_update(self) -> None:
 
         self._attributes = {}
 
-        data = self.api.get_stop_info(self._stop)
+        data: EnturPublicTransportData = self.api.get_stop_info(self._stop)
         if data is None:
             self._state = None
             return

diff --git a/mypy.ini b/mypy.ini
@@ -1348,9 +1348,6 @@ ignore_errors = true
 [mypy-homeassistant.components.enphase_envoy.*]
 ignore_errors = true
 
-[mypy-homeassistant.components.entur_public_transport.*]
-ignore_errors = true
-
 [mypy-homeassistant.components.evohome.*]
 ignore_errors = true
 

diff --git a/script/hassfest/mypy_config.py b/script/hassfest/mypy_config.py
@@ -27,7 +27,6 @@
     "homeassistant.components.dhcp.*",
     "homeassistant.components.doorbird.*",
     "homeassistant.components.enphase_envoy.*",
-    "homeassistant.components.entur_public_transport.*",
     "homeassistant.components.evohome.*",
     "homeassistant.components.fireservicerota.*",
     "homeassistant.components.firmata.*",