homeassistant-stubs 2024.7.3__py3-none-any.whl → 2024.8.0__py3-none-any.whl

homeassistant-stubs/components/overkiz/sensor.pyi

@@ -17,7 +17,7 @@ from pyoverkiz.types import StateType as OverkizStateType
 @dataclass(frozen=True)
 class OverkizSensorDescription(SensorEntityDescription):
     native_value: Callable[[OverkizStateType], StateType] | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, native_value) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., native_value=...) -> None: ...
 
 SENSOR_DESCRIPTIONS: list[OverkizSensorDescription]
 SUPPORTED_STATES: Incomplete

homeassistant-stubs/components/overkiz/switch.pyi

@@ -19,7 +19,7 @@ class OverkizSwitchDescription(SwitchEntityDescription):
     is_on: Callable[[Callable[[str], OverkizStateType]], bool] | None = ...
     turn_on_args: OverkizStateType | list[OverkizStateType] | None = ...
     turn_off_args: OverkizStateType | list[OverkizStateType] | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, turn_on, turn_off, is_on, turn_on_args, turn_off_args) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., turn_on, turn_off, is_on=..., turn_on_args=..., turn_off_args=...) -> None: ...
 
 SWITCH_DESCRIPTIONS: list[OverkizSwitchDescription]
 SUPPORTED_DEVICES: Incomplete

homeassistant-stubs/components/p1_monitor/coordinator.pyi

@@ -5,7 +5,7 @@ from homeassistant.const import CONF_HOST as CONF_HOST
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from p1monitor import Phases, Settings, SmartMeter, WaterMeter
+from p1monitor import Phases as Phases, Settings as Settings, SmartMeter as SmartMeter, WaterMeter as WaterMeter
 from typing import TypedDict
 
 class P1MonitorData(TypedDict):

homeassistant-stubs/components/peco/sensor.pyi

@@ -16,7 +16,7 @@ from typing import Final
 class PECOSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[PECOCoordinatorData], int | str]
     attribute_fn: Callable[[PECOCoordinatorData], dict[str, str]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn, attribute_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn, attribute_fn) -> None: ...
 
 PARALLEL_UPDATES: Final[int]
 SENSOR_LIST: tuple[PECOSensorEntityDescription, ...]

homeassistant-stubs/components/persistent_notification/__init__.pyi

@@ -26,10 +26,10 @@ class Notification(TypedDict):
     title: str | None
 
 class UpdateType(StrEnum):
-    CURRENT: str
-    ADDED: str
-    REMOVED: str
-    UPDATED: str
+    CURRENT = 'current'
+    ADDED = 'added'
+    REMOVED = 'removed'
+    UPDATED = 'updated'
 
 SIGNAL_PERSISTENT_NOTIFICATIONS_UPDATED: Incomplete
 SCHEMA_SERVICE_NOTIFICATION: Incomplete

homeassistant-stubs/components/pi_hole/binary_sensor.pyi

@@ -14,7 +14,7 @@ from typing import Any
 class PiHoleBinarySensorEntityDescription(BinarySensorEntityDescription):
     state_value: Callable[[Hole], bool]
     extra_value: Callable[[Hole], dict[str, Any] | None] = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, state_value, extra_value) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., state_value, extra_value=...) -> None: ...
 
 BINARY_SENSOR_TYPES: tuple[PiHoleBinarySensorEntityDescription, ...]
 

homeassistant-stubs/components/pi_hole/update.pyi

@@ -15,7 +15,7 @@ class PiHoleUpdateEntityDescription(UpdateEntityDescription):
     latest_version: Callable[[dict], str | None] = ...
     release_base_url: str | None = ...
     title: str | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, installed_version, latest_version, release_base_url, title) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., installed_version=..., latest_version=..., release_base_url=..., title=...) -> None: ...
 
 UPDATE_ENTITY_TYPES: tuple[PiHoleUpdateEntityDescription, ...]
 

homeassistant-stubs/components/ping/entity.pyi

@@ -1,7 +1,6 @@
 from .coordinator import PingUpdateCoordinator as PingUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.core import DOMAIN as DOMAIN
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 

homeassistant-stubs/components/ping/sensor.pyi

@@ -13,7 +13,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 class PingSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[PingResult], float | None]
     has_fn: Callable[[PingResult], bool]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn, has_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn, has_fn) -> None: ...
 
 SENSORS: tuple[PingSensorEntityDescription, ...]
 

homeassistant-stubs/components/plugwise/binary_sensor.pyi

@@ -16,7 +16,7 @@ SEVERITIES: Incomplete
 @dataclass(frozen=True)
 class PlugwiseBinarySensorEntityDescription(BinarySensorEntityDescription):
     key: BinarySensorType
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=...) -> None: ...
 
 BINARY_SENSORS: tuple[PlugwiseBinarySensorEntityDescription, ...]
 

homeassistant-stubs/components/plugwise/number.pyi

@@ -13,7 +13,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 @dataclass(frozen=True, kw_only=True)
 class PlugwiseNumberEntityDescription(NumberEntityDescription):
     key: NumberType
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., max_value=..., min_value=..., mode=..., native_max_value=..., native_min_value=..., native_step=..., native_unit_of_measurement=..., step=...) -> None: ...
 
 NUMBER_TYPES: Incomplete
 

homeassistant-stubs/components/plugwise/select.pyi

@@ -14,7 +14,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 class PlugwiseSelectEntityDescription(SelectEntityDescription):
     key: SelectType
     options_key: SelectOptionsType
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, options, options_key) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., options=..., options_key) -> None: ...
 
 SELECT_TYPES: Incomplete
 

homeassistant-stubs/components/plugwise/sensor.pyi

@@ -12,7 +12,7 @@ from plugwise.constants import SensorType as SensorType
 @dataclass(frozen=True)
 class PlugwiseSensorEntityDescription(SensorEntityDescription):
     key: SensorType
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=...) -> None: ...
 
 SENSORS: tuple[PlugwiseSensorEntityDescription, ...]
 

homeassistant-stubs/components/plugwise/switch.pyi

@@ -14,7 +14,7 @@ from typing import Any
 @dataclass(frozen=True)
 class PlugwiseSwitchEntityDescription(SwitchEntityDescription):
     key: SwitchType
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=...) -> None: ...
 
 SWITCHES: tuple[PlugwiseSwitchEntityDescription, ...]
 

homeassistant-stubs/components/powerwall/__init__.pyi

@@ -1,5 +1,5 @@
 from .const import DOMAIN as DOMAIN, POWERWALL_API_CHANGED as POWERWALL_API_CHANGED, POWERWALL_COORDINATOR as POWERWALL_COORDINATOR, UPDATE_INTERVAL as UPDATE_INTERVAL
-from .models import PowerwallBaseInfo as PowerwallBaseInfo, PowerwallData as PowerwallData, PowerwallRuntimeData as PowerwallRuntimeData
+from .models import PowerwallBaseInfo as PowerwallBaseInfo, PowerwallConfigEntry as PowerwallConfigEntry, PowerwallData as PowerwallData, PowerwallRuntimeData as PowerwallRuntimeData
 from _typeshed import Incomplete
 from homeassistant.components import persistent_notification as persistent_notification
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -30,11 +30,11 @@ class PowerwallDataManager:
     async def async_update_data(self) -> PowerwallData: ...
     async def _update_data(self) -> PowerwallData: ...
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_migrate_entity_unique_ids(hass: HomeAssistant, entry: ConfigEntry, base_info: PowerwallBaseInfo) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: PowerwallConfigEntry) -> bool: ...
+async def async_migrate_entity_unique_ids(hass: HomeAssistant, entry: PowerwallConfigEntry, base_info: PowerwallBaseInfo) -> None: ...
 async def _login_and_fetch_base_info(power_wall: Powerwall, host: str, password: str | None) -> PowerwallBaseInfo: ...
 async def _call_base_info(power_wall: Powerwall, host: str) -> PowerwallBaseInfo: ...
 async def get_backup_reserve_percentage(power_wall: Powerwall) -> float | None: ...
 async def _fetch_powerwall_data(power_wall: Powerwall) -> PowerwallData: ...
-def async_last_update_was_successful(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+def async_last_update_was_successful(hass: HomeAssistant, entry: PowerwallConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/powerwall/binary_sensor.pyi

@@ -1,15 +1,13 @@
-from .const import DOMAIN as DOMAIN
 from .entity import PowerWallEntity as PowerWallEntity
-from .models import PowerwallRuntimeData as PowerwallRuntimeData
+from .models import PowerwallConfigEntry as PowerwallConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
 CONNECTED_GRID_STATUSES: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: PowerwallConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class PowerWallRunningSensor(PowerWallEntity, BinarySensorEntity):
     _attr_translation_key: str

homeassistant-stubs/components/powerwall/models.pyi

@@ -1,8 +1,11 @@
 from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from tesla_powerwall import BatteryResponse as BatteryResponse, DeviceType as DeviceType, GridStatus as GridStatus, MetersAggregatesResponse as MetersAggregatesResponse, Powerwall, PowerwallStatusResponse as PowerwallStatusResponse, SiteInfoResponse as SiteInfoResponse, SiteMasterResponse as SiteMasterResponse
+from tesla_powerwall import BatteryResponse as BatteryResponse, DeviceType as DeviceType, GridStatus as GridStatus, MetersAggregatesResponse as MetersAggregatesResponse, Powerwall as Powerwall, PowerwallStatusResponse as PowerwallStatusResponse, SiteInfoResponse as SiteInfoResponse, SiteMasterResponse as SiteMasterResponse
 from typing import TypedDict
 
+PowerwallConfigEntry = ConfigEntry[PowerwallRuntimeData]
+
 @dataclass
 class PowerwallBaseInfo:
     gateway_din: str

homeassistant-stubs/components/powerwall/sensor.pyi

@@ -1,11 +1,10 @@
-from .const import DOMAIN as DOMAIN, POWERWALL_COORDINATOR as POWERWALL_COORDINATOR
+from .const import POWERWALL_COORDINATOR as POWERWALL_COORDINATOR
 from .entity import BatteryEntity as BatteryEntity, PowerWallEntity as PowerWallEntity
-from .models import BatteryResponse as BatteryResponse, PowerwallRuntimeData as PowerwallRuntimeData
+from .models import BatteryResponse as BatteryResponse, PowerwallConfigEntry as PowerwallConfigEntry, PowerwallRuntimeData as PowerwallRuntimeData
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfFrequency as UnitOfFrequency, UnitOfPower as UnitOfPower
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity import Entity as Entity
@@ -21,11 +20,10 @@ _ValueT = TypeVar('_ValueT', bound=float | int | str | None)
 @dataclass(frozen=True, kw_only=True)
 class PowerwallSensorEntityDescription(SensorEntityDescription, Generic[_ValueParamT, _ValueT]):
     value_fn: Callable[[_ValueParamT], _ValueT]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 def _get_meter_power(meter: MeterResponse) -> float: ...
 def _get_meter_frequency(meter: MeterResponse) -> float: ...
-def _get_meter_total_current(meter: MeterResponse) -> float: ...
 def _get_meter_average_voltage(meter: MeterResponse) -> float: ...
 
 POWERWALL_INSTANT_SENSORS: Incomplete
@@ -36,7 +34,7 @@ def _get_instant_current(battery: BatteryResponse) -> float | None: ...
 
 BATTERY_INSTANT_SENSORS: list[PowerwallSensorEntityDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: PowerwallConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class PowerWallChargeSensor(PowerWallEntity, SensorEntity):
     _attr_translation_key: str

homeassistant-stubs/components/powerwall/switch.pyi

@@ -1,9 +1,7 @@
-from .const import DOMAIN as DOMAIN
 from .entity import PowerWallEntity as PowerWallEntity
-from .models import PowerwallRuntimeData as PowerwallRuntimeData
+from .models import PowerwallConfigEntry as PowerwallConfigEntry, PowerwallRuntimeData as PowerwallRuntimeData
 from _typeshed import Incomplete
 from homeassistant.components.switch import SwitchDeviceClass as SwitchDeviceClass, SwitchEntity as SwitchEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -13,7 +11,7 @@ from typing import Any
 
 OFF_GRID_STATUSES: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: PowerwallConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class PowerwallOffGridEnabledEntity(PowerWallEntity, SwitchEntity):
     _attr_translation_key: str

homeassistant-stubs/components/private_ble_device/sensor.pyi

@@ -12,7 +12,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 @dataclass(frozen=True, kw_only=True)
 class PrivateDeviceSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[HomeAssistant, bluetooth.BluetoothServiceInfoBleak], str | int | float | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 SENSOR_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/prometheus/__init__.pyi

@@ -1,8 +1,9 @@
 from _typeshed import Incomplete
 from aiohttp import web
 from collections.abc import Callable as Callable
-from homeassistant.components.climate import ATTR_CURRENT_TEMPERATURE as ATTR_CURRENT_TEMPERATURE, ATTR_HVAC_ACTION as ATTR_HVAC_ACTION, ATTR_HVAC_MODES as ATTR_HVAC_MODES, ATTR_TARGET_TEMP_HIGH as ATTR_TARGET_TEMP_HIGH, ATTR_TARGET_TEMP_LOW as ATTR_TARGET_TEMP_LOW, HVACAction as HVACAction
+from homeassistant.components.climate import ATTR_CURRENT_TEMPERATURE as ATTR_CURRENT_TEMPERATURE, ATTR_FAN_MODE as ATTR_FAN_MODE, ATTR_FAN_MODES as ATTR_FAN_MODES, ATTR_HVAC_ACTION as ATTR_HVAC_ACTION, ATTR_HVAC_MODES as ATTR_HVAC_MODES, ATTR_TARGET_TEMP_HIGH as ATTR_TARGET_TEMP_HIGH, ATTR_TARGET_TEMP_LOW as ATTR_TARGET_TEMP_LOW, HVACAction as HVACAction
 from homeassistant.components.cover import ATTR_CURRENT_POSITION as ATTR_CURRENT_POSITION, ATTR_CURRENT_TILT_POSITION as ATTR_CURRENT_TILT_POSITION
+from homeassistant.components.fan import ATTR_DIRECTION as ATTR_DIRECTION, ATTR_OSCILLATING as ATTR_OSCILLATING, ATTR_PERCENTAGE as ATTR_PERCENTAGE, ATTR_PRESET_MODE as ATTR_PRESET_MODE, ATTR_PRESET_MODES as ATTR_PRESET_MODES, DIRECTION_FORWARD as DIRECTION_FORWARD, DIRECTION_REVERSE as DIRECTION_REVERSE
 from homeassistant.components.http import HomeAssistantView as HomeAssistantView, KEY_HASS as KEY_HASS
 from homeassistant.components.humidifier import ATTR_AVAILABLE_MODES as ATTR_AVAILABLE_MODES, ATTR_HUMIDITY as ATTR_HUMIDITY
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS
@@ -84,6 +85,7 @@ class PrometheusMetrics:
     @staticmethod
     def _unit_string(unit: str | None) -> str | None: ...
     def _handle_switch(self, state: State) -> None: ...
+    def _handle_fan(self, state: State) -> None: ...
     def _handle_zwave(self, state: State) -> None: ...
     def _handle_automation(self, state: State) -> None: ...
     def _handle_counter(self, state: State) -> None: ...

homeassistant-stubs/components/prusalink/__init__.pyi

@@ -1,6 +1,6 @@
 from .config_flow import ConfigFlow as ConfigFlow
 from .const import DOMAIN as DOMAIN
-from .coordinator import JobUpdateCoordinator as JobUpdateCoordinator, LegacyStatusCoordinator as LegacyStatusCoordinator, PrusaLinkUpdateCoordinator as PrusaLinkUpdateCoordinator, StatusCoordinator as StatusCoordinator
+from .coordinator import InfoUpdateCoordinator as InfoUpdateCoordinator, JobUpdateCoordinator as JobUpdateCoordinator, LegacyStatusCoordinator as LegacyStatusCoordinator, PrusaLinkUpdateCoordinator as PrusaLinkUpdateCoordinator, StatusCoordinator as StatusCoordinator
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/components/prusalink/binary_sensor.pyi

@@ -0,0 +1,36 @@
+from . import PrusaLinkEntity as PrusaLinkEntity
+from .const import DOMAIN as DOMAIN
+from .coordinator import PrusaLinkUpdateCoordinator as PrusaLinkUpdateCoordinator
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.binary_sensor import BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from pyprusalink.types import JobInfo, PrinterInfo, PrinterStatus
+from pyprusalink.types_legacy import LegacyPrinterStatus
+from typing import Generic, TypeVar
+
+T = TypeVar('T', PrinterStatus, LegacyPrinterStatus, JobInfo, PrinterInfo)
+
+@dataclass(frozen=True)
+class PrusaLinkBinarySensorEntityDescriptionMixin(Generic[T]):
+    value_fn: Callable[[T], bool]
+    def __init__(self, value_fn) -> None: ...
+
+@dataclass(frozen=True)
+class PrusaLinkBinarySensorEntityDescription(BinarySensorEntityDescription, PrusaLinkBinarySensorEntityDescriptionMixin[T], Generic[T]):
+    available_fn: Callable[[T], bool] = ...
+    def __init__(self, value_fn, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., available_fn=...) -> None: ...
+
+BINARY_SENSORS: dict[str, tuple[PrusaLinkBinarySensorEntityDescription, ...]]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class PrusaLinkBinarySensorEntity(PrusaLinkEntity, BinarySensorEntity):
+    entity_description: PrusaLinkBinarySensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: PrusaLinkUpdateCoordinator, description: PrusaLinkBinarySensorEntityDescription) -> None: ...
+    @property
+    def is_on(self) -> bool: ...

homeassistant-stubs/components/prusalink/button.pyi

@@ -22,7 +22,7 @@ class PrusaLinkButtonEntityDescriptionMixin(Generic[T]):
 @dataclass(frozen=True)
 class PrusaLinkButtonEntityDescription(ButtonEntityDescription, PrusaLinkButtonEntityDescriptionMixin[T], Generic[T]):
     available_fn: Callable[[T], bool] = ...
-    def __init__(self, press_fn, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, available_fn) -> None: ...
+    def __init__(self, press_fn, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., available_fn=...) -> None: ...
 
 BUTTONS: dict[str, tuple[PrusaLinkButtonEntityDescription, ...]]
 

homeassistant-stubs/components/prusalink/coordinator.pyi

@@ -6,7 +6,7 @@ from datetime import timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from pyprusalink import JobInfo, LegacyPrinterStatus, PrinterStatus, PrusaLink as PrusaLink
+from pyprusalink import JobInfo, LegacyPrinterStatus, PrinterInfo, PrinterStatus, PrusaLink as PrusaLink
 from typing import TypeVar
 
 _LOGGER: Incomplete
@@ -32,3 +32,6 @@ class LegacyStatusCoordinator(PrusaLinkUpdateCoordinator[LegacyPrinterStatus]):
 
 class JobUpdateCoordinator(PrusaLinkUpdateCoordinator[JobInfo]):
     async def _fetch_data(self) -> JobInfo: ...
+
+class InfoUpdateCoordinator(PrusaLinkUpdateCoordinator[PrinterInfo]):
+    async def _fetch_data(self) -> PrinterInfo: ...

homeassistant-stubs/components/prusalink/sensor.pyi

@@ -13,11 +13,11 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 from homeassistant.helpers.typing import StateType as StateType
 from homeassistant.util.dt import utcnow as utcnow
 from homeassistant.util.variance import ignore_variance as ignore_variance
-from pyprusalink.types import JobInfo, PrinterStatus
+from pyprusalink.types import JobInfo, PrinterInfo, PrinterStatus
 from pyprusalink.types_legacy import LegacyPrinterStatus
 from typing import Generic, TypeVar
 
-T = TypeVar('T', PrinterStatus, LegacyPrinterStatus, JobInfo)
+T = TypeVar('T', PrinterStatus, LegacyPrinterStatus, JobInfo, PrinterInfo)
 
 @dataclass(frozen=True)
 class PrusaLinkSensorEntityDescriptionMixin(Generic[T]):
@@ -27,7 +27,7 @@ class PrusaLinkSensorEntityDescriptionMixin(Generic[T]):
 @dataclass(frozen=True)
 class PrusaLinkSensorEntityDescription(SensorEntityDescription, PrusaLinkSensorEntityDescriptionMixin[T], Generic[T]):
     available_fn: Callable[[T], bool] = ...
-    def __init__(self, value_fn, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, available_fn) -> None: ...
+    def __init__(self, value_fn, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., available_fn=...) -> None: ...
 
 SENSORS: dict[str, tuple[PrusaLinkSensorEntityDescription, ...]]
 

homeassistant-stubs/components/pure_energie/sensor.pyi

@@ -14,7 +14,7 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity as Coordi
 @dataclass(frozen=True, kw_only=True)
 class PureEnergieSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[PureEnergieData], int | float]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 SENSORS: tuple[PureEnergieSensorEntityDescription, ...]
 

homeassistant-stubs/components/purpleair/config_flow.pyi

@@ -31,7 +31,7 @@ def async_get_remove_sensor_schema(sensors: list[SelectOptionDict]) -> vol.Schem
 class ValidationResult:
     data: Any = ...
     errors: dict[str, Any] = ...
-    def __init__(self, data, errors) -> None: ...
+    def __init__(self, data=..., errors=...) -> None: ...
 
 async def async_validate_api_key(hass: HomeAssistant, api_key: str) -> ValidationResult: ...
 async def async_validate_coordinates(hass: HomeAssistant, api_key: str, latitude: float, longitude: float, distance: float) -> ValidationResult: ...

homeassistant-stubs/components/purpleair/sensor.pyi

@@ -16,7 +16,7 @@ CONCENTRATION_PARTICLES_PER_100_MILLILITERS: Incomplete
 @dataclass(frozen=True, kw_only=True)
 class PurpleAirSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[SensorModel], float | str | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 SENSOR_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/pvoutput/sensor.pyi

@@ -15,7 +15,7 @@ from pvo import Status as Status, System as System
 @dataclass(frozen=True, kw_only=True)
 class PVOutputSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[Status], int | float | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 SENSORS: tuple[PVOutputSensorEntityDescription, ...]
 

homeassistant-stubs/components/qnap_qsw/binary_sensor.pyi

@@ -16,7 +16,7 @@ class QswBinarySensorEntityDescription(BinarySensorEntityDescription, QswEntityD
     attributes: dict[str, list[str]] | None = ...
     qsw_type: QswEntityType | None = ...
     sep_key: str = ...
-    def __init__(self, subkey, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, attributes, qsw_type, sep_key) -> None: ...
+    def __init__(self, subkey, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., attributes=..., qsw_type=..., sep_key=...) -> None: ...
 
 BINARY_SENSOR_TYPES: Final[tuple[QswBinarySensorEntityDescription, ...]]
 LACP_PORT_BINARY_SENSOR_TYPES: Final[tuple[QswBinarySensorEntityDescription, ...]]

homeassistant-stubs/components/qnap_qsw/button.pyi

@@ -15,7 +15,7 @@ from typing import Final
 @dataclass(frozen=True, kw_only=True)
 class QswButtonDescription(ButtonEntityDescription):
     press_action: Callable[[QnapQswApi], Awaitable[bool]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, press_action) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., press_action) -> None: ...
 
 BUTTON_TYPES: Final[tuple[QswButtonDescription, ...]]
 

homeassistant-stubs/components/qnap_qsw/entity.pyi

@@ -31,7 +31,7 @@ class QswEntityDescriptionMixin:
 @dataclass(frozen=True)
 class QswEntityDescription(EntityDescription, QswEntityDescriptionMixin):
     attributes: dict[str, list[str]] | None = ...
-    def __init__(self, subkey, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, attributes) -> None: ...
+    def __init__(self, subkey, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., attributes=...) -> None: ...
 
 class QswSensorEntity(QswDataEntity):
     entity_description: QswEntityDescription

homeassistant-stubs/components/qnap_qsw/sensor.pyi

@@ -16,7 +16,7 @@ class QswSensorEntityDescription(SensorEntityDescription, QswEntityDescription):
     attributes: dict[str, list[str]] | None = ...
     qsw_type: QswEntityType | None = ...
     sep_key: str = ...
-    def __init__(self, subkey, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, attributes, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, qsw_type, sep_key) -> None: ...
+    def __init__(self, subkey, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., attributes=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., qsw_type=..., sep_key=...) -> None: ...
 
 SENSOR_TYPES: Final[tuple[QswSensorEntityDescription, ...]]
 LACP_PORT_SENSOR_TYPES: Final[tuple[QswSensorEntityDescription, ...]]

homeassistant-stubs/components/rabbitair/fan.pyi

@@ -19,6 +19,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 
 class RabbitAirFanEntity(RabbitAirBaseEntity, FanEntity):
     _attr_supported_features: Incomplete
+    _enable_turn_on_off_backwards_compatibility: bool
     _attr_preset_modes: Incomplete
     _attr_speed_count: Incomplete
     def __init__(self, coordinator: RabbitAirDataUpdateCoordinator, entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/radarr/coordinator.pyi

@@ -23,7 +23,7 @@ class RadarrEventMixIn:
 
 @dataclass
 class RadarrEvent(CalendarEvent, RadarrEventMixIn):
-    def __init__(self, release_type, start, end, summary, description, location, uid, recurrence_id, rrule) -> None: ...
+    def __init__(self, release_type, start, end, summary, description=..., location=..., uid=..., recurrence_id=..., rrule=...) -> None: ...
 
 class RadarrDataUpdateCoordinator(DataUpdateCoordinator[T], ABC, Generic[T], metaclass=abc.ABCMeta):
     config_entry: RadarrConfigEntry

homeassistant-stubs/components/radarr/sensor.pyi

@@ -22,7 +22,7 @@ class RadarrSensorEntityDescriptionMixIn(Generic[T]):
 @dataclasses.dataclass(frozen=True)
 class RadarrSensorEntityDescription(SensorEntityDescription, RadarrSensorEntityDescriptionMixIn[T], Generic[T]):
     description_fn: Callable[[RadarrSensorEntityDescription[T], RootFolder], tuple[RadarrSensorEntityDescription[T], str] | None] | None = ...
-    def __init__(self, value_fn, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, description_fn) -> None: ...
+    def __init__(self, value_fn, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., description_fn=...) -> None: ...
 
 SENSOR_TYPES: dict[str, RadarrSensorEntityDescription[Any]]
 BYTE_SIZES: Incomplete

homeassistant-stubs/components/rainforest_raven/sensor.pyi

@@ -14,7 +14,7 @@ from typing import Any
 class RAVEnSensorEntityDescription(SensorEntityDescription):
     message_key: str
     attribute_keys: list[str] | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, message_key, attribute_keys) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., message_key, attribute_keys=...) -> None: ...
 
 SENSORS: Incomplete
 DIAGNOSTICS: Incomplete

homeassistant-stubs/components/rainmachine/__init__.pyi

@@ -57,6 +57,7 @@ SERVICE_PUSH_FLOW_METER_DATA_SCHEMA: Incomplete
 SERVICE_PUSH_WEATHER_DATA_SCHEMA: Incomplete
 SERVICE_RESTRICT_WATERING_SCHEMA: Incomplete
 COORDINATOR_UPDATE_INTERVAL_MAP: Incomplete
+RainMachineConfigEntry = ConfigEntry[RainMachineData]
 
 @dataclass
 class RainMachineData:
@@ -64,12 +65,12 @@ class RainMachineData:
     coordinators: dict[str, RainMachineDataUpdateCoordinator]
     def __init__(self, controller, coordinators) -> None: ...
 
-def async_get_entry_for_service_call(hass: HomeAssistant, call: ServiceCall) -> ConfigEntry: ...
-async def async_update_programs_and_zones(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_reload_entry(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+def async_get_entry_for_service_call(hass: HomeAssistant, call: ServiceCall) -> RainMachineConfigEntry: ...
+async def async_update_programs_and_zones(hass: HomeAssistant, entry: RainMachineConfigEntry) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> bool: ...
+async def async_migrate_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> bool: ...
+async def async_reload_entry(hass: HomeAssistant, entry: RainMachineConfigEntry) -> None: ...
 
 class RainMachineEntity(CoordinatorEntity[RainMachineDataUpdateCoordinator]):
     _attr_has_entity_name: bool
@@ -79,7 +80,7 @@ class RainMachineEntity(CoordinatorEntity[RainMachineDataUpdateCoordinator]):
     _data: Incomplete
     _version_coordinator: Incomplete
     entity_description: Incomplete
-    def __init__(self, entry: ConfigEntry, data: RainMachineData, description: RainMachineEntityDescription) -> None: ...
+    def __init__(self, entry: RainMachineConfigEntry, data: RainMachineData, description: RainMachineEntityDescription) -> None: ...
     @property
     def device_info(self) -> DeviceInfo: ...
     def _handle_coordinator_update(self) -> None: ...