From c702ffa7dd3bf7e5a3edc44d6e1686310ef280e9 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 31 Jul 2024 20:48:30 +0200
Subject: [PATCH] Fix implicit-return in uk_transport (#122932)

---
 homeassistant/components/uk_transport/sensor.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/uk_transport/sensor.py b/homeassistant/components/uk_transport/sensor.py
index 8e874be0bca..a86f7a1cc83 100644
--- a/homeassistant/components/uk_transport/sensor.py
+++ b/homeassistant/components/uk_transport/sensor.py
@@ -6,6 +6,7 @@ from datetime import datetime, timedelta
 from http import HTTPStatus
 import logging
 import re
+from typing import Any
 
 import requests
 import voluptuous as vol
@@ -196,10 +197,10 @@ class UkTransportLiveBusTimeSensor(UkTransportSensor):
                 self._state = None
 
     @property
-    def extra_state_attributes(self):
+    def extra_state_attributes(self) -> dict[str, Any] | None:
         """Return other details about the sensor state."""
-        attrs = {}
         if self._data is not None:
+            attrs = {ATTR_NEXT_BUSES: self._next_buses}
             for key in (
                 ATTR_ATCOCODE,
                 ATTR_LOCALITY,
@@ -207,8 +208,8 @@ class UkTransportLiveBusTimeSensor(UkTransportSensor):
                 ATTR_REQUEST_TIME,
             ):
                 attrs[key] = self._data.get(key)
-            attrs[ATTR_NEXT_BUSES] = self._next_buses
             return attrs
+        return None
 
 
 class UkTransportLiveTrainTimeSensor(UkTransportSensor):
@@ -266,15 +267,17 @@ class UkTransportLiveTrainTimeSensor(UkTransportSensor):
                     self._state = None
 
     @property
-    def extra_state_attributes(self):
+    def extra_state_attributes(self) -> dict[str, Any] | None:
         """Return other details about the sensor state."""
-        attrs = {}
         if self._data is not None:
-            attrs[ATTR_STATION_CODE] = self._station_code
-            attrs[ATTR_CALLING_AT] = self._calling_at
+            attrs = {
+                ATTR_STATION_CODE: self._station_code,
+                ATTR_CALLING_AT: self._calling_at,
+            }
             if self._next_trains:
                 attrs[ATTR_NEXT_TRAINS] = self._next_trains
             return attrs
+        return None
 
 
 def _delta_mins(hhmm_time_str):