diff --git a/custom_components/jablotron100/alarm_control_panel.py b/custom_components/jablotron100/alarm_control_panel.py index 99e29d2..98b58f8 100644 --- a/custom_components/jablotron100/alarm_control_panel.py +++ b/custom_components/jablotron100/alarm_control_panel.py @@ -106,10 +106,10 @@ async def _arm_partially(self, state: StateType, code: str | None = None) -> Non def _detect_supported_features(self) -> AlarmControlPanelEntityFeature: partially_arming_mode = self._jablotron.partially_arming_mode() - if (partially_arming_mode == PartiallyArmingMode.NOT_SUPPORTED): + if partially_arming_mode == PartiallyArmingMode.NOT_SUPPORTED: return AlarmControlPanelEntityFeature.ARM_AWAY - if (partially_arming_mode == PartiallyArmingMode.HOME_MODE): + if partially_arming_mode == PartiallyArmingMode.HOME_MODE: return AlarmControlPanelEntityFeature.ARM_AWAY | AlarmControlPanelEntityFeature.ARM_HOME return AlarmControlPanelEntityFeature.ARM_AWAY | AlarmControlPanelEntityFeature.ARM_NIGHT