Update likelihoods.py

JaxGaussianProcesses · Mar 22, 2024 · d33849d · d33849d
1 parent 700a406
commit d33849d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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.