Fix transmission Coordinator typing (#101903)

This commit is contained in:
Marc Mueller 2023-10-12 22:13:33 +02:00 committed by GitHub
parent 6d2fbeb556
commit 8b134f26a9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 5 deletions

View File

@ -4,7 +4,6 @@ from __future__ import annotations
from contextlib import suppress from contextlib import suppress
from typing import Any from typing import Any
from transmission_rpc.session import SessionStats
from transmission_rpc.torrent import Torrent from transmission_rpc.torrent import Torrent
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
@ -92,7 +91,9 @@ async def async_setup_entry(
async_add_entities(dev, True) async_add_entities(dev, True)
class TransmissionSensor(CoordinatorEntity[SessionStats], SensorEntity): class TransmissionSensor(
CoordinatorEntity[TransmissionDataUpdateCoordinator], SensorEntity
):
"""A base class for all Transmission sensors.""" """A base class for all Transmission sensors."""
_attr_has_entity_name = True _attr_has_entity_name = True

View File

@ -3,8 +3,6 @@ from collections.abc import Callable
import logging import logging
from typing import Any from typing import Any
from transmission_rpc.session import SessionStats
from homeassistant.components.switch import SwitchEntity from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME from homeassistant.const import CONF_NAME
@ -38,7 +36,9 @@ async def async_setup_entry(
async_add_entities(dev, True) async_add_entities(dev, True)
class TransmissionSwitch(CoordinatorEntity[SessionStats], SwitchEntity): class TransmissionSwitch(
CoordinatorEntity[TransmissionDataUpdateCoordinator], SwitchEntity
):
"""Representation of a Transmission switch.""" """Representation of a Transmission switch."""
_attr_has_entity_name = True _attr_has_entity_name = True