diff --git a/src/MaxText/input_pipeline/synthetic_data_processing.py b/src/MaxText/input_pipeline/synthetic_data_processing.py
index 852c839dd..e0107d937 100644
--- a/src/MaxText/input_pipeline/synthetic_data_processing.py
+++ b/src/MaxText/input_pipeline/synthetic_data_processing.py
@@ -94,6 +94,9 @@ def __iter__(self):
   def __next__(self):
     return next(self.data_generator)
 
+  def reset(self):
+    pass
+
   @staticmethod
   def get_place_holder_synthetic_data(config: pyconfig.HyperParameters):
     """fill negative value in synthetic data"""