diff --git a/naive-nlu/tree_nlu/parsing.py b/naive-nlu/tree_nlu/parsing.py index 13436bd..265bd59 100644 --- a/naive-nlu/tree_nlu/parsing.py +++ b/naive-nlu/tree_nlu/parsing.py @@ -130,6 +130,8 @@ def apply_remix(tokens, remix): rebuilt = [] for i in remix: if isinstance(i, int): + if i >= len(tokens): + return None rebuilt.append(tokens[i]) else: assert(isinstance(i, str))