def put_train_fpa()

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

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

def put_train_fpa(root, train_length):
    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'] * train_length)
        print(train_output, file=train_tgt)