code: def reset(self):
      self.file_idx = 0
      self.file_idx_init = True
      self.used = []
  
      self.clean_index = list(
          filter(lambda x: x not in self.used, self.index)
      )
      self.sample_fn = self.sample_once()
  
file_path: mesh-transformer-jax/tfrecord_loader.py
imports: import jax
  import tensorflow
  import numpy
  from transformers import GPT2TokenizerFast
  import itertools
label_desc: Use image node_id 2 for calling the TFRecordLoader obj's underlying member method code with example usage: obj.reset() without return types