From eb2ed89f1d1014ce2336fc3b777a8896ce83f96a Mon Sep 17 00:00:00 2001 From: Alessandro Vullo Date: Tue, 17 Sep 2024 15:08:38 +0100 Subject: [PATCH] Correct another runtime instance error. --- trieste/models/gpflux/models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/trieste/models/gpflux/models.py b/trieste/models/gpflux/models.py index ca75ff163..fbbfd77a4 100644 --- a/trieste/models/gpflux/models.py +++ b/trieste/models/gpflux/models.py @@ -308,7 +308,6 @@ def model_gpflux(self) -> DeepGP: def model_keras(self) -> tf_keras.Model: return self._model_keras - @inherit_check_shapes def sample_encoded(self, query_points: TensorType, num_samples: int) -> TensorType: trajectory = self.trajectory_sampler().get_trajectory() expanded_query_points = tf.expand_dims(query_points, -2) # [N, 1, D]