def _set_checkpoint_dict(self, checkpoint_dict: Dict[(str, Any)]):
    'Sets the checkpoint dict in the task. Only used for testing.\n\n        Args:\n            checkpoint_dict: A serializable dict representing current task state\n        '
    self.checkpoint_dict = checkpoint_dict
    return self
Sets the checkpoint dict in the task. Only used for testing.

Args:
    checkpoint_dict: A serializable dict representing current task state
def _set_checkpoint_dict(self, checkpoint_dict: Dict[(str, Any)]):
    'Sets the checkpoint dict in the task. Only used for testing.\n\n        Args:\n            checkpoint_dict: A serializable dict representing current task state\n        '
    self.checkpoint_dict = checkpoint_dict
    return self