From b21632ad05c2f48aeea36adc96046cfa5afab96d Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Tue, 7 May 2024 18:28:42 +0200
Subject: [PATCH] Improve energy platform typing (#117003)

---
 homeassistant/components/energy/websocket_api.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/energy/websocket_api.py b/homeassistant/components/energy/websocket_api.py
index 2b5b71d3e2f..38cd87a22f5 100644
--- a/homeassistant/components/energy/websocket_api.py
+++ b/homeassistant/components/energy/websocket_api.py
@@ -8,7 +8,6 @@ from collections.abc import Awaitable, Callable
 from datetime import timedelta
 import functools
 from itertools import chain
-from types import ModuleType
 from typing import Any, cast
 
 import voluptuous as vol
@@ -64,13 +63,15 @@ async def async_get_energy_platforms(
 
     @callback
     def _process_energy_platform(
-        hass: HomeAssistant, domain: str, platform: ModuleType
+        hass: HomeAssistant,
+        domain: str,
+        platform: EnergyPlatform,
     ) -> None:
         """Process energy platforms."""
         if not hasattr(platform, "async_get_solar_forecast"):
             return
 
-        platforms[domain] = cast(EnergyPlatform, platform).async_get_solar_forecast
+        platforms[domain] = platform.async_get_solar_forecast
 
     await async_process_integration_platforms(
         hass, DOMAIN, _process_energy_platform, wait_for_platforms=True