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):