def create_input_record()

in 06_preprocessing/jpeg_to_tfrecord_tft.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

def create_input_record(filename, label):
    label_list = label.to_pylist()
    filename_list = filename.to_pylist()
    assert len(filename_list) == 1 and len(filename_list[0]) == 1
    assert len(label_list) == 1 and len(label_list[0]) == 1
    contents = tf.io.read_file(filename_list[0][0]).numpy()
    return {
        'img_bytes': contents,
        'label': label_list[0][0],
        'label_int': LABELS.index(label_list[0][0].decode())
    }