def main()

in tasks/dax-compo/generate_data.py [0:0]

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

def main(train_N):
    try:
        shutil.rmtree(str(train_N))
    except:
        pass

    pathlib.Path(str(train_N)).mkdir()

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

    alphabet_input = string.ascii_lowercase[:train_N]
    alphabet_output = string.ascii_uppercase[:train_N] 

    mem_modifier = lambda _: ' '.join([alphabet_output[0]] * 3)
    mem_prim = lambda x : x
    compo = lambda x: ' '.join([x] * 3)

    generate_mdl(root=f'{train_N}/mem_func/', train_N=train_N, rule=mem_modifier)
    generate_mdl(root=f'{train_N}/mem_prim/', train_N=train_N, rule=mem_prim)
    generate_mdl(root=f'{train_N}/compo/', train_N=train_N, rule=compo)