def get_randaug(aug: str) -> str:
  """Returns a string representing the RandAugment op to use during preprocessing."""
  l, m = {
      'light0': (2, 0),
      'light1': (2, 10),
      'medium1': (2, 15),
      'medium2': (2, 15),
      'strong1': (2, 20),
      'strong2': (2, 20),
      'extreme1': (4, 15),
      'extreme2': (4, 20),
  }[aug]
  return f'randaug({l}, {m})'