diff --git a/homeassistant/components/transmission/sensor.py b/homeassistant/components/transmission/sensor.py index 0b949e73f47..0a0f0dae383 100644 --- a/homeassistant/components/transmission/sensor.py +++ b/homeassistant/components/transmission/sensor.py @@ -4,7 +4,6 @@ from __future__ import annotations from contextlib import suppress from typing import Any -from transmission_rpc.session import SessionStats from transmission_rpc.torrent import Torrent from homeassistant.components.sensor import SensorDeviceClass, SensorEntity @@ -92,7 +91,9 @@ async def async_setup_entry( async_add_entities(dev, True) -class TransmissionSensor(CoordinatorEntity[SessionStats], SensorEntity): +class TransmissionSensor( + CoordinatorEntity[TransmissionDataUpdateCoordinator], SensorEntity +): """A base class for all Transmission sensors.""" _attr_has_entity_name = True diff --git a/homeassistant/components/transmission/switch.py b/homeassistant/components/transmission/switch.py index 253ceb558b9..3e7573b1951 100644 --- a/homeassistant/components/transmission/switch.py +++ b/homeassistant/components/transmission/switch.py @@ -3,8 +3,6 @@ from collections.abc import Callable import logging from typing import Any -from transmission_rpc.session import SessionStats - from homeassistant.components.switch import SwitchEntity from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_NAME @@ -38,7 +36,9 @@ async def async_setup_entry( async_add_entities(dev, True) -class TransmissionSwitch(CoordinatorEntity[SessionStats], SwitchEntity): +class TransmissionSwitch( + CoordinatorEntity[TransmissionDataUpdateCoordinator], SwitchEntity +): """Representation of a Transmission switch.""" _attr_has_entity_name = True