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.