Changes according to review

music-assistant · Jan 2, 2024 · a250bca · a250bca
1 parent bb057d0
commit a250bca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/mass/config_flow.py b/custom_components/mass/config_flow.py
@@ -350,11 +350,11 @@ def async_get_options_flow(config_entry):
 class OptionsFlowHandler(config_entries.OptionsFlow):
     """Class to handle options flow."""
 
-    def __init__(self, config_entry: config_entries.ConfigEntry):
+    def __init__(self, config_entry: config_entries.ConfigEntry) -> None:
         """Initialize options flow."""
         self.config_entry = config_entry
 
-    async def async_step_init(self, user_input=None):
+    async def async_step_init(self, user_input=None) -> FlowResult:
         """Manage the options."""
         LOGGER.debug(
             "OptionsFlowHandler:async_step_init user_input [%s] data [%s]",
@@ -381,7 +381,7 @@ async def async_step_init(self, user_input=None):
             data_schema=vol.Schema(schema),
         )
 
-    def mass_config_option_schema(self, config_entry: config_entries.ConfigEntry) -> dict:
+    def mass_config_option_schema(self, config_entry: config_entries.ConfigEntry) -> vol.Schema:
         """Return a schema for MusicAssistant completion options."""
         return {
             vol.Required(