pass **kwargs to ApproximateGP.__call__ in DeepGPLayer (#2224)

Co-authored-by: root <root@com120226.localdomain> Co-authored-by: Geoff Pleiss <gpleiss@gmail.com>
cornellius-gp · Jan 4, 2023 · 3595191 · 3595191
1 parent e80e5cd
commit 3595191
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpytorch/models/deep_gps/deep_gp.py b/gpytorch/models/deep_gps/deep_gp.py
@@ -97,7 +97,7 @@ def __call__(self, inputs, are_samples=False, **kwargs):
             inputs = inputs.expand(*inputs.shape[:-3], self.output_dims, *inputs.shape[-2:])
 
         # Now run samples through the GP
-        output = ApproximateGP.__call__(self, inputs)
+        output = ApproximateGP.__call__(self, inputs, **kwargs)
         if self.output_dims is not None:
             mean = output.loc.transpose(-1, -2)
             covar = BlockDiagLinearOperator(output.lazy_covariance_matrix, block_dim=-3)