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]