def main()

in tasks/count-or-mem/generate_data.py [0:0]

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

def main(train_length, test_span):
    try:
        shutil.rmtree(str(train_length))
    except:
        pass

    pathlib.Path(str(train_length)).mkdir()

    generate_fpa(root=f'{train_length}/fpa/', train_length=train_length, test_span=test_span)

    memorization = lambda _: train_length
    count = lambda x: x

    generate_mdl(root=f'{train_length}/mem/', train_length=train_length, test_span=test_span, rule=memorization)
    generate_mdl(root=f'{train_length}/count/', train_length=train_length, test_span=test_span, rule=count)