Merge def9f8c into 82fe0b1

Graviti-AI · Dec 27, 2021 · 5e59ae2 · 5e59ae2
2 parents 82fe0b1 + def9f8c
commit 5e59ae2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorbay/client/job.py b/tensorbay/client/job.py
@@ -151,10 +151,10 @@ def update(self, until_complete: bool = False) -> None:
                 sleep(_JOB_UPDATE_INTERVAL)
                 job_info = self._job_updater(self.job_id)
 
-        self.finished_at = job_info["finishedAt"]
+        self.finished_at = job_info.get("finishedAt")
         self.status = job_info["status"]
         self.error_message = job_info["errorMessage"]
-        self._result = job_info["results"]
+        self._result = job_info.get("results")
 
     def abort(self) -> None:
         """Abort a :class:`Job`."""