From ec430a5cbac0b0db73c6a1e3d4e3cde127ad60ac Mon Sep 17 00:00:00 2001 From: Simone Margaritelli Date: Thu, 24 Oct 2019 19:54:55 +0200 Subject: [PATCH] misc: small fix or general refactoring i did not bother commenting --- pwnagotchi/ai/epoch.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pwnagotchi/ai/epoch.py b/pwnagotchi/ai/epoch.py index 23b1d95c..8f5380a5 100644 --- a/pwnagotchi/ai/epoch.py +++ b/pwnagotchi/ai/epoch.py @@ -83,12 +83,13 @@ class Epoch(object): bond_unit_scale = self.config['personality']['bond_encounters_factor'] self.num_peers = len(peers) + num_peers = self.num_peers + 1e-10 # avoid division by 0 + self.tot_bond_factor = sum((peer.encounters for peer in peers)) / bond_unit_scale - self.avg_bond_factor = self.tot_bond_factor / self.num_peers + self.avg_bond_factor = self.tot_bond_factor / num_peers num_aps = len(aps) + 1e-10 num_sta = sum(len(ap['clients']) for ap in aps) + 1e-10 - num_peers = self.num_peers + 1e-10 # avoid division by 0 aps_per_chan = [0.0] * wifi.NumChannels sta_per_chan = [0.0] * wifi.NumChannels peers_per_chan = [0.0] * wifi.NumChannels