def get_num_records()

in benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py [0:0]

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

def get_num_records(filenames):
    def count_records(tf_record_filename):
        count = 0
        for _ in tf.python_io.tf_record_iterator(tf_record_filename):
            count += 1
        return count

    nfile = len(filenames)
    return count_records(filenames[0]) * (nfile - 1) + count_records(filenames[-1])