Fixes Saving Format related test failures for Keras 3 (#2180)

* Deprecate keras_v3 save format for Keras3

* Deprecate keras_v3 save format for Keras3

* Deprecate keras_v3 save format for Keras3

Author: sampathweb

keras_cv/models/segmentation/deeplab_v3_plus/deeplab_v3_plus_test.py

@@ -21,6 +21,7 @@
 
 from keras_cv.backend import keras
 from keras_cv.backend import ops
+from keras_cv.backend.config import keras_3
 from keras_cv.models import DeepLabV3Plus
 from keras_cv.models import ResNet18V2Backbone
 from keras_cv.models.backbones.test_backbone_presets import (
@@ -86,12 +87,8 @@ def test_with_model_preset_forward_pass(self):
         expected_output = np.zeros((1, 512, 512, 1))
         self.assertAllClose(output, expected_output)
 
-    @parameterized.named_parameters(
-        ("tf_format", "tf", "model"),
-        ("keras_format", "keras_v3", "model.keras"),
-    )
     @pytest.mark.large  # Saving is slow, so mark these large.
-    def test_saved_model(self, save_format, filename):
+    def test_saved_model(self):
         target_size = [512, 512, 3]
 
         backbone = ResNet18V2Backbone(input_shape=target_size)
@@ -100,8 +97,11 @@ def test_saved_model(self, save_format, filename):
         input_batch = np.ones(shape=[2] + target_size)
         model_output = model(input_batch)
 
-        save_path = os.path.join(self.get_temp_dir(), filename)
-        model.save(save_path, save_format=save_format)
+        save_path = os.path.join(self.get_temp_dir(), "model.keras")
+        if keras_3():
+            model.save(save_path)
+        else:
+            model.save(save_path, save_format="keras_v3")
         restored_model = keras.models.load_model(save_path)
 
         # Check we got the real object back.

keras_cv/models/segmentation/segformer/segformer_test.py

@@ -20,6 +20,7 @@
 
 from keras_cv.backend import keras
 from keras_cv.backend import ops
+from keras_cv.backend.config import keras_3
 from keras_cv.models import MiTBackbone
 from keras_cv.models import SegFormer
 from keras_cv.tests.test_case import TestCase
@@ -81,7 +82,10 @@ def test_saved_model(self):
         model_output = model(input_batch)
 
         save_path = os.path.join(self.get_temp_dir(), "model.keras")
-        model.save(save_path, save_format="keras_v3")
+        if keras_3():
+            model.save(save_path)
+        else:
+            model.save(save_path, save_format="keras_v3")
         restored_model = keras.models.load_model(save_path)
 
         # Check we got the real object back.

keras_cv/models/segmentation/segment_anything/sam_test.py

@@ -22,6 +22,7 @@
 
 from keras_cv.backend import keras
 from keras_cv.backend import ops
+from keras_cv.backend.config import keras_3
 from keras_cv.models.backbones.vit_det.vit_det_aliases import ViTDetBBackbone
 from keras_cv.models.segmentation.segment_anything.sam import (
     SegmentAnythingModel,
@@ -282,7 +283,10 @@ def test_end_to_end_model_save(self):
 
         # Save the model
         save_path = os.path.join(self.get_temp_dir(), "model.keras")
-        model.save(save_path, save_format="keras_v3")
+        if keras_3():
+            model.save(save_path)
+        else:
+            model.save(save_path, save_format="keras_v3")
         restored_model = keras.models.load_model(save_path)
 
         # Check we got the real object back.