Internal change

PiperOrigin-RevId: 707299915
google · Dec 18, 2024 · 723f839 · 723f839
1 parent 316cd52
commit 723f839
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/edward2/tensorflow/layers/gaussian_process.py b/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)