def get_structure_task(
    config: ml_collections.ConfigDict, is_train: bool, task_id: str
) -> Task:
  """Get frame from flow task setup."""
  del is_train
  task_config = ml_collections.FrozenConfigDict(
      config.get(task_id, ml_collections.ConfigDict()))
  task_token = SpecialTokens.TASK_STRUCTURE
  mask_fn = functools.partial(mask_utils.all_mask)
  # Currently the cond video is computed directly in token_dumper because the
  # cond_fns only receive the input rgb, but we also need flow. Therefore return
  # None here.
  cond_fn = None
  return Task(task_id, task_token, mask_fn, cond_fn, task_config)
Get frame from flow task setup.
videogvt/train_lib/task_registry.py