def put_train_mdl()

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

• 16 lines of code
• 3 McCabe index (conditional complexity)

def put_train_mdl(root, train_N, rule):

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

    with open(f'{root}/train.src', 'w') as train_src, open(f'{root}/train.dst', 'w') as train_tgt:
        # Primitives
        for train_input, train_output in zip(alphabet_input, alphabet_output):
            print(train_input, file=train_src)
            print(train_output, file=train_tgt)

        # Function
        train_input = 'F ' + alphabet_input[0]
        print(train_input, file=train_src)
        train_output = ' '.join([alphabet_output[0]] * 3)
        print(train_output, file=train_tgt)

        for char_inp, char_out in zip(alphabet_input[1:], alphabet_output[1:]):
            train_input = 'F ' + char_inp
            print(train_input, file=train_src)
            train_output = rule(char_out)
            print(train_output, file=train_tgt)