Bug fix to account for state space

8 years ago · 6e81c34a29
parent 85e3c7f6df
commit 6e81c34a29
1 changed files with 4 additions and 3 deletions
--- a/src/api/index.py
+++ b/src/api/index.py
@ -53,9 +53,10 @@ def procs(id):
 		data =  gReader.read()
 		ahandler = AnalyzeAnomaly()
 		learn = {}
-		for row in data['learn'] :
-			label = row['label']
-			learn[label] = row
+		if 'learn' in data :
+			for row in data['learn'] :
+				label = row['label']
+				learn[label] = row
 		r = {}
 		for label in data :
 			if label not in ['learn','folders'] :