Fix __repr__ for tfd.GaussianProcess when index_points is None.

PiperOrigin-RevId: 464690048

Author: emilyfertig

tensorflow_probability/python/distributions/gaussian_process.py

@@ -774,6 +774,22 @@ def _type_spec(self):
   def _convert_variables_to_tensors(self):
     return auto_composite_tensor.convert_variables_to_tensors(self)
 
+  def __repr__(self):
+    if self.index_points is None:
+      event_shape_str = '?'
+    else:
+      event_shape_str = distribution._str_tensorshape(self.event_shape)
+    return ('<tfp.distributions.{type_name} '
+            '\'{self_name}\''
+            ' batch_shape={batch_shape}'
+            ' event_shape={event_shape}'
+            ' dtype={dtype}>'.format(
+                type_name=type(self).__name__,
+                self_name=self.name or '<unknown>',
+                batch_shape=distribution._str_tensorshape(self.batch_shape),
+                event_shape=event_shape_str,
+                dtype=distribution._str_dtype(self.dtype)))
+
 
 @auto_composite_tensor.type_spec_register(
     'tfp.distributions.GaussianProcess_ACTTypeSpec')

tensorflow_probability/python/distributions/gaussian_process_test.py

@@ -213,6 +213,9 @@ def _kernel_fn(x, y):
     with self.assertRaises(ValueError):
       gp.mean()
 
+    self.assertIn("event_shape=?", repr(gp))
+    self.assertIn("event_shape=[10]", repr(gp.copy(index_points=index_points)))
+
   def testMarginalHasCorrectTypes(self):
     gp = tfd.GaussianProcess(
         kernel=psd_kernels.ExponentiatedQuadratic(), validate_args=True)