Reset `_num_batches_produced` counter for the reset() method so that it properly resets for TFDS.
Reset `_num_batches_produced` counter for the reset() method so that it properly resets for TFDS.

PiperOrigin-RevId: 415061727

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo