diff --git a/naive-nlu/tree_nlu/parsing.py b/naive-nlu/tree_nlu/parsing.py index a43f5f1..d539a28 100644 --- a/naive-nlu/tree_nlu/parsing.py +++ b/naive-nlu/tree_nlu/parsing.py @@ -298,12 +298,10 @@ def get_similar_tree(knowledge_base, atom, tokens): return None for i, possibility in enumerate(sorted_possibilities): - with session().log(possibility): - similar_matcher, similar_result, similar_result_resolved, _, _ = possibility - session().annotate('AST: {}'.format(similar_result)) - session().annotate('Based on: {}'.format(similar_matcher)) + similar_matcher, similar_result, similar_result_resolved, _, _ = possibility + with session().log("Like {}".format(similar_matcher)): session().annotate('Results on: {}'.format(similar_result_resolved)) - session().annotate('---------------------') + session().annotate('AST: {}'.format(similar_result)) return sorted_possibilities[0]