Forgot to add parametrization to test

Author: gianlucasama

Diff for: keras_cv/models/object_detection/retinanet/retinanet_test.py

@@ -244,7 +244,12 @@ def test_saved_model(self, save_format, filename):
         restored_output = restored_model(input_batch)
         self.assertAllClose(model_output, restored_output)
 
-    @pytest.mark.large
+    @parameterized.named_parameters(
+        ("tf_format", "tf", "model"),
+        ("h5_format", "h5", "model.h5"),
+        ("keras_format", "keras_v3", "model.keras"),
+    )
+    @pytest.mark.large  # Saving is slow, so mark these large.
     def test_custom_saved_model(self, save_format, filename):
         class CustomPredictionHead(
             keras_cv.models.object_detection.retinanet.PredictionHead