Improve typing in ihc util

homeassistant/components/ihc/util.py

@@ -2,34 +2,44 @@
 
 import asyncio
 
-from homeassistant.core import callback
+from ihcsdk.ihccontroller import IHCController
 
+from homeassistant.core import HomeAssistant, callback
 
-async def async_pulse(hass, ihc_controller, ihc_id: int):
+
+async def async_pulse(
+    hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int
+) -> None:
     """Send a short on/off pulse to an IHC controller resource."""
     await async_set_bool(hass, ihc_controller, ihc_id, True)
     await asyncio.sleep(0.1)
     await async_set_bool(hass, ihc_controller, ihc_id, False)
 
 
 @callback
-def async_set_bool(hass, ihc_controller, ihc_id: int, value: bool):
+def async_set_bool(
+    hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int, value: bool
+) -> asyncio.Future[bool]:
     """Set a bool value on an IHC controller resource."""
     return hass.async_add_executor_job(
         ihc_controller.set_runtime_value_bool, ihc_id, value
     )
 
 
 @callback
-def async_set_int(hass, ihc_controller, ihc_id: int, value: int):
+def async_set_int(
+    hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int, value: int
+) -> asyncio.Future[bool]:
     """Set a int value on an IHC controller resource."""
     return hass.async_add_executor_job(
         ihc_controller.set_runtime_value_int, ihc_id, value
     )
 
 
 @callback
-def async_set_float(hass, ihc_controller, ihc_id: int, value: float):
+def async_set_float(
+    hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int, value: float
+) -> asyncio.Future[bool]:
     """Set a float value on an IHC controller resource."""
     return hass.async_add_executor_job(
         ihc_controller.set_runtime_value_float, ihc_id, value