Fix 'learning empty word' bug

sergioburdisso · Feb 27, 2020 · 0d3d7e1 · 0d3d7e1
1 parent 52fbf45
commit 0d3d7e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyss3/__init__.py b/pyss3/__init__.py
@@ -1806,7 +1806,7 @@ def learn(self, doc, cat, n_grams=1, prep=True, update=True):
         Print.info("learning...", offset=1)
         tips = []
         for word in doc:
-            if word != ".":
+            if word and word != '.':
                 self.__prun_counter__ += 1
                 # if word doesn't exist yet, then...
                 try: