also use observation_noise without mask in ModelList

pytorch · Feb 6, 2025 · b7d0570 · b7d0570
1 parent a43bd4d
commit b7d0570
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/botorch/models/model.py b/botorch/models/model.py
@@ -661,6 +661,8 @@ def fantasize(
                 sampler_i = (
                     sampler.samplers[i] if isinstance(sampler, ListSampler) else sampler
                 )
+                if observation_noise is not None:
+                    observation_noise_i = observation_noise[..., i : i + 1]
 
             fant_model = self.models[i].fantasize(
                 X=X_i,