From 6c46f9db4b18de0be31e06d4fcb9e98cc5a9d3d2 Mon Sep 17 00:00:00 2001 From: kenkeiras Date: Sun, 15 Apr 2018 20:46:30 +0200 Subject: [PATCH] Fix element_matches_bugs when element is a dictionary. --- naive-nlu/tree_nlu/parsing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/naive-nlu/tree_nlu/parsing.py b/naive-nlu/tree_nlu/parsing.py index b43084e..b06e18b 100644 --- a/naive-nlu/tree_nlu/parsing.py +++ b/naive-nlu/tree_nlu/parsing.py @@ -423,7 +423,7 @@ def all_matching_indexes(knowledge_base, collection, element): def element_matches_groups(knowledge, element: Dict, groups): if isinstance(groups, str) and groups in knowledge: - return len(knowledge[element].get("groups", set()) & element['groups']) > 0 + return len(knowledge[groups].get("groups", set()) & element['groups']) > 0 elif isinstance(groups, dict): return len(element.get("groups", set()) & element['groups']) > 0 return False