feat(api): fixed typiung errors

AntaresSimulatorTeam · Dec 13, 2024 · 9f88c9d · 9f88c9d
1 parent 1d3adec
commit 9f88c9d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/antares/service/api_services/run_api.py b/src/antares/service/api_services/run_api.py
@@ -71,7 +71,7 @@ def wait_job_completion(self, job: Job, time_out: int) -> None:
         if job.status == JobStatus.FAILED:
             raise SimulationFailedError(self.study_id)
 
-        if job.unzip_output:
+        if job.unzip_output and job.output_id:
             self._wait_unzip_output(self.study_id, ["UNARCHIVE"], job.output_id)
 
         return None
@@ -95,6 +95,7 @@ def _get_task_id(self, job_output_id: str, tasks: list[dict[str, Any]]) -> str:
             output_id = task_name.split("/")[-1].split(" ")[0]
             if output_id == job_output_id:
                 return task["id"]
+        raise AntaresSimulationUnzipError(self.study_id, "Could not find task for unarchiving job")
 
     def _get_task_until_success(self, url: str, repeat_interval: int) -> None:
         task_success = False