def _ensure_eval_suffix(benchmark):
  """Returns a string ending in an eval suffix; adds ``'_e'`` suffix if needed.

  Args:
    benchmark: Name of a benchmark or task, that might already include an
        eval-indicating suffix (``'_e'`` or ``'_e2'``).
  """
  if benchmark.endswith('_e') or benchmark.endswith('_e2'):
    return benchmark
  else:
    return benchmark + '_e'
Returns a string ending in an eval suffix; adds ``'_e'`` suffix if needed.

  Args:
    benchmark: Name of a benchmark or task, that might already include an
        eval-indicating suffix (``'_e'`` or ``'_e2'``).
  
trax/data/tf_inputs.py