Skip to content

Commit c6826d9

Browse files
Merge pull request #2522 from rafaol/master
Added optional `kwargs` to `ExactMarginalLogLikelihood` call
2 parents 25da2cc + 21f75fd commit c6826d9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gpytorch/mlls/exact_marginal_log_likelihood.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def _add_other_terms(self, res, params):
4949

5050
return res
5151

52-
def forward(self, function_dist, target, *params):
52+
def forward(self, function_dist, target, *params, **kwargs):
5353
r"""
5454
Computes the MLL given :math:`p(\mathbf f)` and :math:`\mathbf y`.
5555
@@ -63,7 +63,7 @@ def forward(self, function_dist, target, *params):
6363
raise RuntimeError("ExactMarginalLogLikelihood can only operate on Gaussian random variables")
6464

6565
# Determine output likelihood
66-
output = self.likelihood(function_dist, *params)
66+
output = self.likelihood(function_dist, *params, **kwargs)
6767

6868
# Remove NaN values if enabled
6969
if settings.observation_nan_policy.value() == "mask":

0 commit comments

Comments
 (0)