def METHOD_NAME(
    self,
    *,
    hydra_context: HydraContext,
    task_function: TaskFunction,
    config: DictConfig,
) -> None:
    self.config = config
    self.hydra_context = hydra_context
    self.task_function = task_function