evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Round off probability to 2 decimals. (#9365)

Browse source 
* Round off probablity to 2 decimals.

* Update tests

* remove debug print
This commit is contained in:
Alok Saboo 2017-09-12 00:44:51 +05:30 • committed by Pascal Vizeli
parent cc1979691e
commit c7ecebfd07
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
homeassistant/components/binary_sensor/bayesian.py
View file

@ -126,7 +126,6 @@ class BayesianBinarySensor(BinarySensorDevice):
self.watchers[platform](entity_obs)
prior = self.prior
print(self.current_obs.values())
for obs in self.current_obs.values():
prior = update_probability(prior, obs['prob_true'],
obs['prob_false'])
@ -201,7 +200,7 @@ class BayesianBinarySensor(BinarySensorDevice):
"""Return the state attributes of the sensor."""
return {
'observations': [val for val in self.current_obs.values()],
'probability': self.probability,
'probability': round(self.probability, 2),
'probability_threshold': self._probability_threshold
}