@classmethod
def GetProgramSchedule(cls, class_key):
    """Retrieve the ProgramSchedule and a dict of task params.

    Args:
      class_key: String class key (i.e. `image.mnist.LeNet5`).

    Returns:
      ProgramSchedule.Params()
    """
    model_params_cls = cls.GetClass(class_key)
    model_params = model_params_cls()
    program_schedule_cfg = model_params.ProgramSchedule()
    return program_schedule_cfg
