old_contents: # Copyright 2022 The KerasCV Authors
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     https://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  import tensorflow as tf
  from absl.testing import parameterized
  
  from keras_cv import core
  from keras_cv import metrics
  
  
  class SerializationTest(tf.test.TestCase, parameterized.TestCase):
      @parameterized.named_parameters(
          ("COCORecall", metrics.COCORecall, {"class_ids": [0, 1, 2]}),
          (
              "COCOMeanAveragePrecision",
              metrics.COCOMeanAveragePrecision,
              {"class_ids": [0, 1, 2]},
          ),
      )
      def test_layer_serialization(self, metric_cls, init_args):
          metric = metric_cls(**init_args)
          metric_config = metric.get_config()
          reconstructed_metric = metric_cls.from_config(metric_config)
  
          self.assertEqual(metric.get_config(), reconstructed_metric.get_config())
  
new_contents: # Copyright 2022 The KerasCV Authors
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     https://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  import tensorflow as tf
  from absl.testing import parameterized
  
  from keras_cv import metrics
  
  
  class SerializationTest(tf.test.TestCase, parameterized.TestCase):
      @parameterized.named_parameters(
          ("COCORecall", metrics.COCORecall, {"class_ids": [0, 1, 2]}),
          (
              "COCOMeanAveragePrecision",
              metrics.COCOMeanAveragePrecision,
              {"class_ids": [0, 1, 2]},
          ),
      )
      def test_layer_serialization(self, metric_cls, init_args):
          metric = metric_cls(**init_args)
          metric_config = metric.get_config()
          reconstructed_metric = metric_cls.from_config(metric_config)
  
          self.assertEqual(metric.get_config(), reconstructed_metric.get_config())
  
subject: Remove core import