def put_train_mdl()

in tasks/add-or-mul/generate_data.py [0:0]

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

def put_train_mdl(root, train_length, test_from, test_to, rule):
    with open(f'{root}/train.src', 'w') as train_src, open(f'{root}/train.dst', 'w') as train_tgt:
        train_input = ' '.join(['a'] * train_length)
        print(train_input, file=train_src)
        train_output = ' '.join(['b'] * (2 * train_length))
        print(train_output, file=train_tgt)


        for length in range(test_from, test_to, 1):
            if length == train_length: continue

            train_input = ' '.join(['a'] * length)
            print(train_input, file=train_src)
            train_output = ' '.join(['b'] * rule(length))
            print(train_output, file=train_tgt)