From 61da16ed915877f8cdb60cc4b1d00de175305fbe Mon Sep 17 00:00:00 2001
From: Simone Margaritelli <evilsocket@gmail.com>
Date: Mon, 7 Oct 2019 01:31:37 +0200
Subject: [PATCH] fix: when the AI is ready the mode label reports AI
---
pwnagotchi/ai/__init__.py | 2 +-
pwnagotchi/ui/view.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/pwnagotchi/ai/__init__.py b/pwnagotchi/ai/__init__.py
index 359a38c..190025f 100644
--- 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
\ No newline at end of file
diff --git a/pwnagotchi/ui/view.py b/pwnagotchi/ui/view.py
index 78e75b6..5847662 100644
--- 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()