def METHOD_NAME(self, image, transformation, **kwargs):
    contrast_factor = transformation
    output = tf.image.adjust_contrast(
        image, contrast_factor=contrast_factor
    )
    output = tf.clip_by_value(output, 0, 255)
    output.set_shape(image.shape)
    return output