issue: keras-cv MaybeApply in KPL

[innat]

Describe

TF 2.12
KerasCV: 0.5

I've tried to run MaybeApply layer on other layers but it gives error,

augmenter = keras.Sequential(
    layers=[
        preprocessor,

        keras_cv.layers.MaybeApply(
            keras_cv.layers.RandomFlip(mode="horizontal", bounding_box_format="xywh"),
            rate=0.7
        ),

        keras_cv.layers.MaybeApply(
            keras_cv.layers.RandomChoice(
                layers=[
                    keras_cv.layers.ChannelShuffle(),
                    keras_cv.layers.RandomChannelShift(value_range=(0, 255), factor=0.2),
                ]),
            rate=0.5
        )
    ]
)

ValueError: Either both boxes and classes should be Ragged, or neither should be ragged. Got type(boxes)=<class 'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensor'>, type(classes)=<class 'tensorflow.python.framework.ops.Tensor'>.

Gist.