diff --git a/datasets/geometric_sampler.py b/datasets/geometric_sampler.py
index 79ba8cf..a0662ea 100644
--- a/datasets/geometric_sampler.py
+++ b/datasets/geometric_sampler.py
@@ -3,6 +3,7 @@
 from clrs._src.specs import SPECS
 class GeometricGraphSampler(Sampler):
     """Bellman-Ford sampler."""
+    CAN_TRUNCATE_INPUT_DATA = True
 
     def _sample_data(
         self,