0066f83bc6f9f3861119db2593c3007796d2056c,deeppavlov/models/seq2seq_go_bot/kb.py,KnowledgeBaseEntityNormalizer,denormalize,#KnowledgeBaseEntityNormalizer#Any#Any#,164

Before Change

    def denormalize(self, key, tokens):
        for entity, value in self.kb([key])[0]:
            if entity in tokens:
                entity_pos = tokens.index(entity)
                // if value is string, split to tokens
                if isinstance(value, str):
                    value = value.split()
                tokens = tokens[:entity_pos] + value + tokens[entity_pos + 1:]
        return tokens

    def __call__(self,
                 keys: List[str],

After Change

    def denormalize(self, tokens, entries):
        for entity, ent_tokens in entries:
            while (entity in tokens):
                ent_pos = tokens.index(entity)
                tokens = tokens[:ent_pos] + ent_tokens + tokens[ent_pos+1:]
        return tokens

    def __call__(self,
                 tokens_list: List[List[str]],

Instances