From 476d04e2fb086f6768ee9a808597f2a68c699a06 Mon Sep 17 00:00:00 2001
From: jan iversen <jancasacondor@gmail.com>
Date: Sat, 18 Sep 2021 11:02:24 +0200
Subject: [PATCH] Activate mypy. (#55965)

---
 .../components/entur_public_transport/sensor.py        | 10 ++++++----
 mypy.ini                                               |  3 ---
 script/hassfest/mypy_config.py                         |  1 -
 3 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/homeassistant/components/entur_public_transport/sensor.py b/homeassistant/components/entur_public_transport/sensor.py
index cad8a49884f2..3256b26171b2 100644
--- 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,9 +160,9 @@ class EnturPublicTransportSensor(SensorEntity):
         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:
@@ -168,7 +170,7 @@ class EnturPublicTransportSensor(SensorEntity):
         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 @@ class EnturPublicTransportSensor(SensorEntity):
 
         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
index 084355c2beb2..fc3d76863ef6 100644
--- 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
index fc6b12ac70ea..6114030f2b22 100644
--- a/script/hassfest/mypy_config.py
+++ b/script/hassfest/mypy_config.py
@@ -27,7 +27,6 @@ IGNORED_MODULES: Final[list[str]] = [
     "homeassistant.components.dhcp.*",
     "homeassistant.components.doorbird.*",
     "homeassistant.components.enphase_envoy.*",
-    "homeassistant.components.entur_public_transport.*",
     "homeassistant.components.evohome.*",
     "homeassistant.components.fireservicerota.*",
     "homeassistant.components.firmata.*",