def train_input_fn()

in src/train.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

def train_input_fn(hvd, mpi=False):
    if mpi:
        return _input(args.epochs, args.batch_size, None, 'train', hvd)
    else:
        return _input(args.epochs, args.batch_size, args.train, 'train')