@utils.map_over_dataset
def _constant_preprocessor(unused_x, val):
    return {'targets': tf.constant([val], tf.int32), 'targets_pretokenized': tf.constant(f'targets_{val}'), 'inputs': tf.constant([val], tf.int32), 'inputs_pretokenized': tf.constant(f'inputs_{val}')}
@utils.map_over_dataset
def _constant_preprocessor(unused_x, val):
    return {'targets': tf.constant([val], tf.int32), 'targets_pretokenized': tf.constant(f'targets_{val}'), 'inputs': tf.constant([val], tf.int32), 'inputs_pretokenized': tf.constant(f'inputs_{val}')}