Fix config_flow.py for HA <2024.4.0b0 (#9)

Fixes #7 Signed-off-by: rany <ranygh@riseup.net>
rany2 · Jun 9, 2024 · a44fa01 · a44fa01
1 parent bb4e014
commit a44fa01
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/open_meteo_solar_forecast/config_flow.py b/custom_components/open_meteo_solar_forecast/config_flow.py
@@ -8,7 +8,6 @@
 from homeassistant.config_entries import (
     ConfigEntry,
     ConfigFlow,
-    ConfigFlowResult,
     OptionsFlow,
 )
 from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME
@@ -25,6 +24,11 @@
     DOMAIN,
 )
 
+try:
+    from homeassistant.config_entries import ConfigFlowResult  # >=2024.4.0b0
+except ImportError:
+    from homeassistant.data_entry_flow import FlowResult as ConfigFlowResult
+
 
 class OpenMeteoSolarForecastFlowHandler(ConfigFlow, domain=DOMAIN):
     """Handle a config flow for Open-Meteo Solar Forecast."""