From 0931a8c10ab6c726b5f451b504a05b512a28bd9d Mon Sep 17 00:00:00 2001 From: Jeroen Oudshoorn Date: Fri, 5 Jan 2024 14:07:18 +0100 Subject: [PATCH] Update --- pwnagotchi/agent.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pwnagotchi/agent.py b/pwnagotchi/agent.py index e5715f4d..178a4acb 100644 --- a/pwnagotchi/agent.py +++ b/pwnagotchi/agent.py @@ -14,12 +14,12 @@ from pwnagotchi.automata import Automata from pwnagotchi.log import LastSession from pwnagotchi.bettercap import Client from pwnagotchi.mesh.utils import AsyncAdvertiser -# from pwnagotchi.ai.train import AsyncTrainer +from pwnagotchi.ai.train import AsyncTrainer RECOVERY_DATA_FILE = '/root/.pwnagotchi-recovery' -class Agent(Client, Automata, AsyncAdvertiser): +class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer): def __init__(self, view, config, keypair): Client.__init__(self, config['bettercap']['hostname'], config['bettercap']['scheme'], @@ -28,7 +28,7 @@ class Agent(Client, Automata, AsyncAdvertiser): config['bettercap']['password']) Automata.__init__(self, config, view) AsyncAdvertiser.__init__(self, config, view, keypair) - # AsyncTrainer.__init__(self, config) + AsyncTrainer.__init__(self, config) self._started_at = time.time() self._filter = None if not config['main']['filter'] else re.compile(config['main']['filter']) @@ -129,7 +129,7 @@ class Agent(Client, Automata, AsyncAdvertiser): time.sleep(1) def start(self): - # self.start_ai() + self.start_ai() self._wait_bettercap() self.setup_events() self.set_starting()