Merge pull request #23 from dadav/fix/empty_brain

Empty brain leads to crash
2019-09-29 21:21:06 +02:00 · 2019-09-29 21:21:06 +02:00 · 91db2feace
commit 91db2feace
parent ac7aaf40ae e35a64734f
1 changed files with 1 additions and 1 deletions
--- a/sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ai/train.py
+++ b/sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ai/train.py
@ -54,7 +54,7 @@ class Stats(object):

    def load(self):
        with self._lock:
-            if os.path.exists(self.path):
+            if os.path.exists(self.path) and os.path.getsize(self.path) > 0:
                core.log("[ai] loading %s" % self.path)
                with open(self.path, 'rt') as fp:
                    obj = json.load(fp)