Register the predicted noise

ziatdinovmax · Jan 12, 2024 · 85fd93b · 85fd93b
1 parent e2a0151
commit 85fd93b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpax/models/mngp.py b/gpax/models/mngp.py
@@ -171,7 +171,7 @@ def predict(
             noise_predicted = self.noise_predicted
         else:
             noise_predicted = self.linreg(self.X_train, self.measured_noise, X_new)
-
+            self.noise_predicted = noise_predicted
         if samples is None:
             samples = self.get_samples(chain_dim=False)
         if device: