diff --git a/keras_cv/layers/serialization_test.py b/keras_cv/layers/serialization_test.py index e216378a0a..6896aeab79 100644 --- a/keras_cv/layers/serialization_test.py +++ b/keras_cv/layers/serialization_test.py @@ -377,6 +377,14 @@ class SerializationTest(TestCase): "from_logits": True, }, ), + ( + "NonMaxSuppression", + cv_layers.NonMaxSuppression, + { + "bounding_box_format": "yxyx", + "from_logits": True, + }, + ), ) def test_layer_serialization(self, layer_cls, init_args): # TODO: Some layers are not yet compatible with Keras 3.