fix typo

ziatdinovmax · Aug 24, 2023 · 218c0a7 · 218c0a7
1 parent b4fc9f0
commit 218c0a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpax/models/vidkl.py b/gpax/models/vidkl.py
@@ -202,7 +202,7 @@ def get_mvn_posterior(self,
         """
         if y_residual is None:
             y_residual = self.y_train
-        noise = params["noise"]
+        noise = k_params["noise"]
         noise_p = noise * (1 - jnp.array(noiseless, int))
         # embed data into the latent space
         z_train = self.nn_module.apply(