Correct another runtime instance error.

secondmind-labs · Sep 17, 2024 · eb2ed89 · eb2ed89
1 parent 6b23bf7
commit eb2ed89
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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]