code: def repeat(dataset: Iterable[_T]) -> Iterator[_T]:
      return itertools.cycle(dataset)
  
file_path: dm-haiku/examples/transformer/dataset.py
imports: from collections.abc import Iterable, Iterator
  import itertools
  import random
  from typing import NamedTuple, TypeVar
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: repeat(dataset) and returns: itertools