Creates non-breaking changes where necessary in preparation for switching all of Keras to new serialization format.

PiperOrigin-RevId: 507864605

Author: nkovela1

edward2/tensorflow/constraints.py

@@ -78,15 +78,16 @@ def get_config(self):
 
 
 def serialize(initializer):
-  return tf.keras.utils.serialize_keras_object(initializer)
+  return tf.keras.utils.legacy.serialize_keras_object(initializer)
 
 
 def deserialize(config, custom_objects=None):
-  return tf.keras.utils.deserialize_keras_object(
+  return tf.keras.utils.legacy.deserialize_keras_object(
       config,
       module_objects=globals(),
       custom_objects=custom_objects,
-      printable_module_name='constraints')
+      printable_module_name='constraints',
+  )
 
 
 def get(identifier, value=None):

edward2/tensorflow/initializers.py

@@ -851,15 +851,16 @@ def get_config(self):
 
 
 def serialize(initializer):
-  return tf.keras.utils.serialize_keras_object(initializer)
+  return tf.keras.utils.legacy.serialize_keras_object(initializer)
 
 
 def deserialize(config, custom_objects=None):
-  return tf.keras.utils.deserialize_keras_object(
+  return tf.keras.utils.legacy.deserialize_keras_object(
       config,
       module_objects=globals(),
       custom_objects=custom_objects,
-      printable_module_name='initializers')
+      printable_module_name='initializers',
+  )
 
 
 def get(identifier, value=None):

edward2/tensorflow/layers/gaussian_process.py

@@ -104,7 +104,7 @@ def get_config(self):
     return {
         'variance': self.variance,
         'bias': self.bias,
-        'encoder': tf.keras.utils.serialize_keras_object(self.encoder),
+        'encoder': tf.keras.utils.legacy.serialize_keras_object(self.encoder),
     }
 
 
@@ -250,9 +250,10 @@ def compute_output_shape(self, input_shape):
   def get_config(self):
     config = {
         'units': self.units,
-        'mean_fn': tf.keras.utils.serialize_keras_object(self.mean_fn),
-        'covariance_fn': tf.keras.utils.serialize_keras_object(
-            self.covariance_fn),
+        'mean_fn': tf.keras.utils.legacy.serialize_keras_object(self.mean_fn),
+        'covariance_fn': tf.keras.utils.legacy.serialize_keras_object(
+            self.covariance_fn
+        ),
         'conditional_inputs': None,  # don't serialize as it can be large
         'conditional_outputs': None,  # don't serialize as it can be large
     }

edward2/tensorflow/regularizers.py

@@ -388,15 +388,16 @@ def get_config(self):
 
 
 def serialize(initializer):
-  return tf.keras.utils.serialize_keras_object(initializer)
+  return tf.keras.utils.legacy.serialize_keras_object(initializer)
 
 
 def deserialize(config, custom_objects=None):
-  return tf.keras.utils.deserialize_keras_object(
+  return tf.keras.utils.legacy.deserialize_keras_object(
       config,
       module_objects=globals(),
       custom_objects=custom_objects,
-      printable_module_name='regularizers')
+      printable_module_name='regularizers',
+  )
 
 
 def get(identifier, value=None):