def _get_file_names()

in tensorflow_managed_spot_training_checkpointing/generate_cifar10_tfrecords.py [0:0]

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

def _get_file_names():
    """Returns the file names expected to exist in the input_dir."""
    return {
        'train': ['data_batch_%d' % i for i in xrange(1, 5)],
        'validation': ['data_batch_5'],
        'eval': ['test_batch'],
    }