def get_allowed_dataset_names(dataset_name: str,
                              task_id: str) -> Tuple[str, ...]:
  """(dataset_name, task) maps to all possible datasets."""
  # TODO(roadjiang): refactor this function.
  if dataset_name == 'unit_test':
    return ('unit_test',)
  ds_names = [
      t for t in EVAL_DATASET_INFO.keys()
      if t.startswith(dataset_name)
  ]
  ret = []
  for ds_name in ds_names:
    supported_tasks = EVAL_DATASET_INFO[ds_name].get('supported_tasks',
                                                     UNLABELED_DS_TASK_IDS)
    if task_id in supported_tasks:
      ret.append(ds_name)
  return tuple(ret)
(dataset_name, task) maps to all possible datasets.
videogvt/train_lib/eval_utils.py