Correct reference to encode function.

secondmind-labs · Sep 17, 2024 · 6b23bf7 · 6b23bf7
1 parent 89736e8
commit 6b23bf7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/trieste/models/gpflux/sampler.py b/trieste/models/gpflux/sampler.py
@@ -476,7 +476,8 @@ def __init__(self, model: GPfluxPredictor, num_features: int):
             DeepGaussianProcessDecoupledLayer(model, i, num_features)
             for i in range(len(model.model_gpflux.f_layers))
         ]
-        self._encode = lambda x: x if model.encoder is None else model.encoder.encode(x)
+
+        self._encode = lambda x: model.encode(x)
 
     @tf.function
     def __call__(self, x: TensorType) -> TensorType: