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

View File

@ -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