mirror of
https://github.com/home-assistant/core.git
synced 2025-04-24 01:08:12 +00:00
Fix transmission Coordinator typing (#101903)
This commit is contained in:
parent
6d2fbeb556
commit
8b134f26a9
@ -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
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user