Fix a couple of typing issues

loopj · Jan 8, 2025 · 06f3f61 · 06f3f61
1 parent 5cd4a1c
commit 06f3f61
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/custom_components/vantage/config_flow.py b/custom_components/vantage/config_flow.py
@@ -181,9 +181,10 @@ async def async_step_reauth(
         self, _entry_data: dict[str, Any] | None = None
     ) -> ConfigFlowResult:
         """Perform reauth after controller authentication error."""
-        self.reauth_entry = self.hass.config_entries.async_get_entry(
-            self.context["entry_id"]
-        )
+        if not (entry_id := self.context.get("entry_id")):
+            return self.async_abort(reason="unknown")
+
+        self.reauth_entry = self.hass.config_entries.async_get_entry(entry_id)
 
         return await self.async_step_reauth_confirm()
 

diff --git a/custom_components/vantage/sensor.py b/custom_components/vantage/sensor.py
@@ -121,6 +121,8 @@ def __post_init__(self) -> None:
             case "Temperature":
                 self._attr_device_class = SensorDeviceClass.TEMPERATURE
                 self._attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
+            case _:
+                pass
 
     @property
     def native_value(self) -> int | Decimal | None: