def yield_records_for_split()

in 05_create_dataset/jpeg_to_tfrecord.py [0:0]

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

def yield_records_for_split(x, desired_split):
    split, rec = x
    # print(split, desired_split, split == desired_split)
    if split == desired_split:
        yield rec