def METHOD_NAME(self):
    inputs = tf.constant([[[0]], [[1]], [[2]]])
    with self.assertRaisesRegex(
        ValueError, "maximum supported output rank is 2"
    ):
        preprocessing_utils.encode_categorical_inputs(
            inputs, "multi_hot", 4, "float32"
        )