Replace tensorflow.python.keras with keras. tensorflow.python.keras is an old copy and is deprecated.

PiperOrigin-RevId: 485551171

Author: tensorflower-gardener

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/cluster_preserve_integration_test.py

@@ -17,7 +17,6 @@
 import numpy as np
 import tensorflow as tf
 
-from tensorflow.python.keras import keras_parameterized
 from tensorflow_model_optimization.python.core.clustering.keras import cluster
 from tensorflow_model_optimization.python.core.clustering.keras import cluster_config
 from tensorflow_model_optimization.python.core.clustering.keras.experimental import cluster as experimental_cluster
@@ -30,7 +29,6 @@
 layers = tf.keras.layers
 
 
-@keras_parameterized.run_all_keras_modes
 class ClusterPreserveIntegrationTest(tf.test.TestCase, parameterized.TestCase):
 
   def setUp(self):

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/cluster_preserve_quantize_registry_test.py

@@ -14,11 +14,8 @@
 # ==============================================================================
 """Tests for ClusterPreserveQuantizeRegistry."""
 
-from absl.testing import parameterized
-
 import tensorflow as tf
 
-from tensorflow.python.keras import keras_parameterized
 
 from tensorflow_model_optimization.python.core.clustering.keras import clustering_registry
 from tensorflow_model_optimization.python.core.quantization.keras import quantize_config
@@ -29,9 +26,7 @@
 layers = tf.keras.layers
 
 
-@keras_parameterized.run_all_keras_modes
-class ClusterPreserveQuantizeRegistryTest(tf.test.TestCase,
-                                          parameterized.TestCase):
+class ClusterPreserveQuantizeRegistryTest(tf.test.TestCase):
 
   def setUp(self):
     super(ClusterPreserveQuantizeRegistryTest, self).setUp()

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve/prune_preserve_quantize_registry_test.py

@@ -13,11 +13,8 @@
 # limitations under the License.
 # ==============================================================================
 """Tests for PrunePreserveQuantizeRegistry."""
-from absl.testing import parameterized
-
 import tensorflow as tf
 
-from tensorflow.python.keras import keras_parameterized
 from tensorflow_model_optimization.python.core.quantization.keras import quantize_config
 from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.prune_preserve import (
     prune_preserve_quantize_registry,)
@@ -28,9 +25,7 @@
 layers = tf.keras.layers
 
 
-@keras_parameterized.run_all_keras_modes
-class PrunePreserveQuantizeRegistryTest(tf.test.TestCase,
-                                        parameterized.TestCase):
+class PrunePreserveQuantizeRegistryTest(tf.test.TestCase):
 
   def setUp(self):
     super(PrunePreserveQuantizeRegistryTest, self).setUp()