def compute_labels()

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

• 14 lines of code
• 8 McCabe index (conditional complexity)

    def compute_labels(cls, limit=20):
        all_labels = string.ascii_lowercase[:limit]
        train = []
        dev = []
        train_vocab = set()
        for i in range(0, len(all_labels)-5, 2):
            a, b, c, d, e = all_labels[i:i+5]
            for trip in itertools.permutations([a, b, d]):
                train.append(trip)
                train_vocab |= set(trip)
            for trip in itertools.permutations([b, c, e]):
                dev.append(trip)
        dev = [(a, b, c) for a, b, c in dev if a in train_vocab and b in train_vocab and c in train_vocab]
        return train, dev