def yield_records_for_split()

in 06_preprocessing/jpeg_to_tfrecord_tft.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