From d607b2210e3565f9aab081e404a3cc3a78d53a42 Mon Sep 17 00:00:00 2001 From: kenkeiras Date: Sat, 30 Sep 2017 00:54:58 +0200 Subject: [PATCH] Avoid crashing when a fit is not found. --- naive-nlu/tree_nlu/knowledge_base.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/naive-nlu/tree_nlu/knowledge_base.py b/naive-nlu/tree_nlu/knowledge_base.py index 6386ce4..dfb2b51 100644 --- a/naive-nlu/tree_nlu/knowledge_base.py +++ b/naive-nlu/tree_nlu/knowledge_base.py @@ -53,13 +53,16 @@ class KnowledgeBase(object): return knowledge_diff_getter - def process(self, row): row = row.lower() knowledge_before = copy.deepcopy(self.knowledge) logging.debug("\x1b[7;32m> {} \x1b[0m".format(row)) tokens = parsing.to_tokens(row) - tokens, inferred_tree = parsing.get_fit(self, tokens) + fit = parsing.get_fit(self, tokens) + if fit is None: + return None + + tokens, inferred_tree = fit result = knowledge_evaluation.integrate_information(self.knowledge, { "elements": tokens,