def get_states(self, task_id): """Returns the (timestamp, status) tuples of the task or [] if could not replay.""" statuses = CheckpointDispatcher.iter_statuses(self._runner_ckpt(task_id)) try: return [(state.timestamp_ms / 1000.0, state.state) for state in statuses] except CheckpointDispatcher.ErrorRecoveringState: return []