From d33849dbdf2f3fe03f6dffee4153e618ecd7efdf Mon Sep 17 00:00:00 2001 From: Daniel Dodd <daniel_dodd@icloud.com> Date: Fri, 22 Mar 2024 10:50:12 +0000 Subject: [PATCH] Update likelihoods.py --- gpjax/likelihoods.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gpjax/likelihoods.py b/gpjax/likelihoods.py index cbcf690f9..e8ec4faeb 100644 --- a/gpjax/likelihoods.py +++ b/gpjax/likelihoods.py @@ -137,6 +137,7 @@ def __init__( self, num_datapoints: int, obs_stddev: tp.Union[ScalarFloat, Float[Array, "#N"], PositiveReal] = 1.0, + integrator: AbstractIntegrator = AnalyticalGaussianIntegrator(), ): r"""Initializes the Gaussian likelihood. @@ -150,7 +151,7 @@ def __init__( else: self.obs_stddev = PositiveReal(jnp.asarray(obs_stddev)) - super().__init__(num_datapoints, AnalyticalGaussianIntegrator()) + super().__init__(num_datapoints, integrator) def link_function(self, f: Float[Array, "..."]) -> tfd.Normal: r"""The link function of the Gaussian likelihood.