BREAKING CHANGE: Ignore deprecated always_yield_multivariate_normal…

… arg to `tfd.GaussianProcess` and `tfd.GaussianProcessRegressionModel`. PiperOrigin-RevId: 518890632
google · Mar 23, 2023 · 625aeb2 · 625aeb2
1 parent 70974a8
commit 625aeb2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/vizier/_src/jax/models/gaussian_process_model.py b/vizier/_src/jax/models/gaussian_process_model.py
@@ -124,7 +124,7 @@ def __call__(
  index_points=inputs,
  observation_noise_variance=observation_noise_variance,
  validate_args=self._use_tfp_runtime_validation,
-  always_yield_multivariate_normal=True)
+ )
 
 
 class GaussianProcessARDWithCategorical(sp_model.ModelCoroutine):
@@ -257,5 +257,4 @@ def __call__(
  index_points=inputs,
  observation_noise_variance=observation_noise_variance,
  validate_args=self._use_tfp_runtime_validation,
- always_yield_multivariate_normal=True,
  )
diff --git a/vizier/_src/jax/models/hebo_gp_model.py b/vizier/_src/jax/models/hebo_gp_model.py
@@ -143,4 +143,4 @@ def _inverse_constraint_fn(f):
  index_points=inputs,
  observation_noise_variance=observation_noise_variance,
  cholesky_fn=None,
-  always_yield_multivariate_normal=True)
+ )
diff --git a/vizier/_src/jax/models/tuned_gp_models.py b/vizier/_src/jax/models/tuned_gp_models.py
@@ -174,5 +174,4 @@ def __call__(
  index_points=inputs,
  observation_noise_variance=observation_noise_variance,
  cholesky_fn=cholesky_fn,
- always_yield_multivariate_normal=True,
  )