From 3a5309e9a0a1e6fd23085740426ebf07b8f48c13 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 6 Sep 2024 15:20:39 +0200
Subject: [PATCH] Improve config flow type hints in tellduslive (#125299)

---
 .../components/tellduslive/config_flow.py          | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/tellduslive/config_flow.py b/homeassistant/components/tellduslive/config_flow.py
index 6b5e7150d67..3bbb34912f9 100644
--- a/homeassistant/components/tellduslive/config_flow.py
+++ b/homeassistant/components/tellduslive/config_flow.py
@@ -35,14 +35,15 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
 
     VERSION = 1
 
+    _session: Session
+
     def __init__(self) -> None:
         """Init config flow."""
         self._hosts = [CLOUD_NAME]
         self._host = None
-        self._session = None
         self._scan_interval = SCAN_INTERVAL
 
-    def _get_auth_url(self):
+    def _get_auth_url(self) -> str | None:
         self._session = Session(
             public_key=PUBLIC_KEY,
             private_key=NOT_SO_PRIVATE_KEY,
@@ -70,7 +71,9 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
             ),
         )
 
-    async def async_step_auth(self, user_input=None):
+    async def async_step_auth(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
         """Handle the submitted configuration."""
         errors = {}
         if user_input is not None:
@@ -114,7 +117,10 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
             },
         )
 
-    async def async_step_discovery(self, discovery_info):
+    async def async_step_discovery(
+        self,
+        discovery_info: list[str],  # type: ignore[override]
+    ) -> ConfigFlowResult:
         """Run when a Tellstick is discovered."""
         await self._async_handle_discovery_without_unique_id()