From c8f11d65d2a4ddf729107c9a6dabec5bfaabc173 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 11 Aug 2022 00:10:28 +0200 Subject: [PATCH] Improve type hints in demo and mqtt number entity (#76464) --- homeassistant/components/demo/number.py | 2 +- homeassistant/components/mqtt/number.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/demo/number.py b/homeassistant/components/demo/number.py index 17382ab9962..9613aa247fb 100644 --- a/homeassistant/components/demo/number.py +++ b/homeassistant/components/demo/number.py @@ -131,7 +131,7 @@ class DemoNumber(NumberEntity): name=self.name, ) - async def async_set_native_value(self, value): + async def async_set_native_value(self, value: float) -> None: """Update the current value.""" self._attr_native_value = value self.async_write_ha_state() diff --git a/homeassistant/components/mqtt/number.py b/homeassistant/components/mqtt/number.py index fbadd653df7..4e9f237431a 100644 --- a/homeassistant/components/mqtt/number.py +++ b/homeassistant/components/mqtt/number.py @@ -272,7 +272,7 @@ class MqttNumber(MqttEntity, RestoreNumber): return self._config.get(CONF_UNIT_OF_MEASUREMENT) @property - def native_value(self): + def native_value(self) -> float | None: """Return the current value.""" return self._current_number