def get_wiki_extract()

in rtfm/tasks/groups.py [0:0]

• 20 lines of code
• 11 McCabe index (conditional complexity)

    def get_wiki_extract(self):
        labels = []
        words = utils.tokenize(self.get_wiki())
        target_element = self.target_monster.element.describe()
        target_modifiers = set([a for t, a in self.modifier_assignment if t == self.target_monster.element][0])
        target_monsters = set([a for t, a in self.group_assignment if t == self.target_group][0])
        for w in words:
            l = 0
            if w == self.target_group:
                l = 1
            elif w == target_element:
                l = 2
            elif w in target_monsters:
                l = 3
            elif w in target_modifiers:
                l = 4
            labels.append(l)
        if len(labels) < self.max_wiki:
            labels += [255] * (self.max_wiki - len(labels))
        return torch.LongTensor(labels)