From db10c313b5737c08e33b2dccdbf4f6ea01119966 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sun, 3 Apr 2022 22:14:11 +0200
Subject: [PATCH] Add type hints to here_travel_time (#69216)

* Add type hints to here_travel_time

* Adjust
---
 homeassistant/components/here_travel_time/model.py  | 4 ++--
 homeassistant/components/here_travel_time/sensor.py | 1 +
 mypy.ini                                            | 3 ---
 script/hassfest/mypy_config.py                      | 1 -
 4 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/here_travel_time/model.py b/homeassistant/components/here_travel_time/model.py
index 5cea81d1ece..eb85e966edf 100644
--- a/homeassistant/components/here_travel_time/model.py
+++ b/homeassistant/components/here_travel_time/model.py
@@ -31,5 +31,5 @@ class HERETravelTimeConfig:
     travel_mode: str
     route_mode: str
     units: str
-    arrival: time
-    departure: time
+    arrival: time | None
+    departure: time | None
diff --git a/homeassistant/components/here_travel_time/sensor.py b/homeassistant/components/here_travel_time/sensor.py
index a0449f7b5c0..304c49b6bed 100644
--- a/homeassistant/components/here_travel_time/sensor.py
+++ b/homeassistant/components/here_travel_time/sensor.py
@@ -285,6 +285,7 @@ class HERETravelTimeSensor(SensorEntity, CoordinatorEntity):
         """Return the attribution."""
         if self.coordinator.data is not None:
             return self.coordinator.data.get(ATTR_ATTRIBUTION)
+        return None
 
     @property
     def icon(self) -> str:
diff --git a/mypy.ini b/mypy.ini
index 35eea971bcd..fdde33a02d3 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -2689,9 +2689,6 @@ ignore_errors = true
 [mypy-homeassistant.components.hassio.websocket_api]
 ignore_errors = true
 
-[mypy-homeassistant.components.here_travel_time.sensor]
-ignore_errors = true
-
 [mypy-homeassistant.components.home_plus_control]
 ignore_errors = true
 
diff --git a/script/hassfest/mypy_config.py b/script/hassfest/mypy_config.py
index 8bbbe0703a4..da558da83bd 100644
--- a/script/hassfest/mypy_config.py
+++ b/script/hassfest/mypy_config.py
@@ -57,7 +57,6 @@ IGNORED_MODULES: Final[list[str]] = [
     "homeassistant.components.hassio.sensor",
     "homeassistant.components.hassio.system_health",
     "homeassistant.components.hassio.websocket_api",
-    "homeassistant.components.here_travel_time.sensor",
     "homeassistant.components.home_plus_control",
     "homeassistant.components.home_plus_control.api",
     "homeassistant.components.icloud",