From ae1592b27a9239f90365da4d5b8afd8cfd10569a Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Sun, 7 Nov 2021 15:57:34 +0100
Subject: [PATCH] Fix typing issues - wallbox.number (#59286)

---
 homeassistant/components/wallbox/number.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/wallbox/number.py b/homeassistant/components/wallbox/number.py
index 837e880afcb..1bc561cb7c5 100644
--- a/homeassistant/components/wallbox/number.py
+++ b/homeassistant/components/wallbox/number.py
@@ -23,7 +23,7 @@ class WallboxNumberEntityDescription(NumberEntityDescription):
     min_value: float = 0
 
 
-NUMBER_TYPES: dict[str, NumberEntityDescription] = {
+NUMBER_TYPES: dict[str, WallboxNumberEntityDescription] = {
     CONF_MAX_CHARGING_CURRENT_KEY: WallboxNumberEntityDescription(
         key=CONF_MAX_CHARGING_CURRENT_KEY,
         name="Max. Charging Current",
@@ -57,6 +57,8 @@ async def async_setup_entry(hass, config, async_add_entities):
 class WallboxNumber(CoordinatorEntity, NumberEntity):
     """Representation of the Wallbox portal."""
 
+    entity_description: WallboxNumberEntityDescription
+
     def __init__(
         self, coordinator, config, description: WallboxNumberEntityDescription
     ):