def enumerate_attribute_value()

in egg/zoo/compositional_efficiency/dataset.py [0:0]

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

def enumerate_attribute_value(n_attributes, n_values, mode):
    from hashlib import md5

    iters = [range(n_values) for _ in range(n_attributes)]

    data = list(itertools.product(*iters))

    if mode is None:
        return data
    elif mode == "train":
        return [x for x in data if hash(str(x)) % 5 != 0]
    elif mode == "test":
        return [x for x in data if hash(str(x)) % 5 == 0]
    assert False