def METHOD_NAME(image_bytes: bytes, label: int) -> tf.train.Example:
  return tf.train.Example(
      features=tf.train.Features(
          feature={
              KEY_IMAGE_BYTES: tf.train.Feature(
                  bytes_list=tf.train.BytesList(value=[image_bytes])
              ),
              KEY_CLASS: tf.train.Feature(
                  int64_list=tf.train.Int64List(value=[label])
              ),
          }
      )
  )