Internal change

PiperOrigin-RevId: 707299915

Author: Edward2 Team

edward2/tensorflow/layers/gaussian_process.py

@@ -406,9 +406,11 @@ def build(self, input_shape=None):
   def call_weights(self):
     """Calls any weights if the initializer is itself a layer."""
     if isinstance(self.inducing_inputs_initializer, tf.keras.layers.Layer):
+      assert self.conditional_inputs is not None
       self.conditional_inputs = self.inducing_inputs_initializer(
           self.conditional_inputs.shape, self.dtype)
     if isinstance(self.inducing_outputs_initializer, tf.keras.layers.Layer):
+      assert self.conditional_outputs is not None
       self.conditional_outputs = self.inducing_outputs_initializer(
           self.conditional_outputs.shape, self.dtype)