def METHOD_NAME(example):
    """Local processing function for dataset mapping."""
    example["class_vectors"] = tf.cast(
        make_class_vectors(example["track_inds"], example["n_tracks"]),
        tf.float32,
    )
    return example