fix boog

naturalness · Nov 28, 2015 · 718b746 · 718b746
1 parent 9753b33
commit 718b746
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/unnaturalcode/sourceModel.py b/unnaturalcode/sourceModel.py
@@ -69,7 +69,10 @@ def predictLexed(self, lexemes):
     def windowedQuery(self, lexemes, returnWindows=True):
         lastWindowStarts = len(lexemes)-self.windowSize
         if lastWindowStarts < 1:
-          return [(lexemes, self.queryLexed(lexemes))]
+            if returnWindows:
+                return [(lexemes, self.queryLexed(lexemes))]
+            else:
+                return [(False, self.queryLexed(lexemes))]                
         r = []
         for i in range(0,lastWindowStarts+1): # remember range is [)
             end = i+self.windowSize