Additionaly checking reset_for_eval=False before throwing the exhau…

…sted dataset error.

PiperOrigin-RevId: 660839907

paxml/seqio_input.py

@@ -1194,7 +1194,7 @@ def _get_targets(self) -> Mapping[str, NestedMap]:
         if self.ds_ragged_tensor_keys:
           example_orig = next(self.targets_iter_ori)
       except (tf.errors.OutOfRangeError, StopIteration) as exc:
-        if self._num_eval_examples > 0:
+        if self._num_eval_examples > 0 and not self.reset_for_eval:
           raise StopIteration(
               'Exhausted eval data with reset_for_eval=False after'
               f' {num_examples-1} examples (batch_size={self.batch_size})'