Update hskgp.py

ziatdinovmax · Nov 16, 2023 · dc95379 · dc95379
1 parent dffaaab
commit dc95379
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpax/models/hskgp.py b/gpax/models/hskgp.py
@@ -213,7 +213,7 @@ def get_data_var_samples(self):
             else:
                 mean_ = self.noise_mean_fn(self.X_train.squeeze())
             log_var += jnp.log(mean_)
-        return jnp.exp(samples["log_var"])
+        return jnp.exp(log_var)
 
     def _print_summary(self):
         samples = self.get_samples(1)