fix: when the AI is ready the mode label reports AI

2019-10-07 01:31:37 +02:00 · 2019-10-07 01:31:37 +02:00 · 61da16ed91
commit 61da16ed91
parent b26d238f40
2 changed files with 2 additions and 2 deletions
--- a/pwnagotchi/ai/init.py
+++ b/pwnagotchi/ai/init.py
@ -39,4 +39,4 @@ def load(config, agent, epoch, from_disk=True):
        for key, value in config['params'].items():
            logging.info("      %s: %s" % (key, value))

-    return a2c
+    return a2c
--- a/pwnagotchi/ui/view.py
+++ b/pwnagotchi/ui/view.py
@ -153,7 +153,7 @@ class View(object):
        self.set('face', faces.AWAKE)

    def on_ai_ready(self):
-        self.set('mode', '')
+        self.set('mode', '  AI')
        self.set('face', faces.HAPPY)
        self.set('status', self._voice.on_ai_ready())
        self.update()