From b86927a4535be0a5936411a769e036a202d6dddc Mon Sep 17 00:00:00 2001 From: HarvsG <11440490+HarvsG@users.noreply.github.com> Date: Sun, 9 Oct 2022 19:30:38 +0100 Subject: [PATCH] Enable strict typing on Bayesian (#79870) * make bayesian static * no longer private --- .strict-typing | 1 + homeassistant/components/bayesian/binary_sensor.py | 3 ++- mypy.ini | 10 ++++++++++ 3 files changed, 13 insertions(+), 1 deletion(-) diff --git a/.strict-typing b/.strict-typing index 61e9c53609a..e47bb51f173 100644 --- a/.strict-typing +++ b/.strict-typing @@ -64,6 +64,7 @@ homeassistant.components.automation.* homeassistant.components.awair.* homeassistant.components.backup.* homeassistant.components.baf.* +homeassistant.components.bayesian.* homeassistant.components.binary_sensor.* homeassistant.components.bluetooth.* homeassistant.components.bluetooth_tracker.* diff --git a/homeassistant/components/bayesian/binary_sensor.py b/homeassistant/components/bayesian/binary_sensor.py index 1d2674255f9..f3a7d08ffb9 100644 --- a/homeassistant/components/bayesian/binary_sensor.py +++ b/homeassistant/components/bayesian/binary_sensor.py @@ -34,6 +34,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.event import ( TrackTemplate, TrackTemplateResult, + TrackTemplateResultInfo, async_track_state_change_event, async_track_template_result, ) @@ -189,7 +190,7 @@ class BayesianBinarySensor(BinarySensorEntity): self._probability_threshold = probability_threshold self._attr_device_class = device_class self._attr_is_on = False - self._callbacks: list = [] + self._callbacks: list[TrackTemplateResultInfo] = [] self.prior = prior self.probability = prior diff --git a/mypy.ini b/mypy.ini index 309068bf2c1..b96efc4b8c3 100644 --- a/mypy.ini +++ b/mypy.ini @@ -392,6 +392,16 @@ disallow_untyped_defs = true warn_return_any = true warn_unreachable = true +[mypy-homeassistant.components.bayesian.*] +check_untyped_defs = true +disallow_incomplete_defs = true +disallow_subclassing_any = true +disallow_untyped_calls = true +disallow_untyped_decorators = true +disallow_untyped_defs = true +warn_return_any = true +warn_unreachable = true + [mypy-homeassistant.components.binary_sensor.*] check_untyped_defs = true disallow_incomplete_defs = true