Skip to content

Commit bf118a2

Browse files
committed
[skip ci] Fix updating job status
1 parent 5cd87c4 commit bf118a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/trainable_entity_extractor/use_cases/OrchestratorUseCase.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ def add_job(self, distributed_job: DistributedJob) -> None:
2020
self.distributed_jobs.append(distributed_job)
2121

2222
def process_job(self, distributed_job: DistributedJob) -> JobProcessingResult:
23+
self.job_executor.update_job_statuses(distributed_job)
24+
2325
if self.job_executor.is_extractor_cancelled(distributed_job.extraction_identifier):
2426
self._cancel_and_remove_job(distributed_job)
2527
return JobProcessingResult(
@@ -111,8 +113,6 @@ def _process_prediction_job(self, distributed_job: DistributedJob) -> JobProcess
111113
)
112114

113115
def _process_performance_job(self, distributed_job: DistributedJob) -> JobProcessingResult:
114-
self.job_executor.update_job_statuses(distributed_job)
115-
116116
if len(distributed_job.sub_jobs) == [x for x in distributed_job.sub_jobs if x.status == JobStatus.WAITING]:
117117
self.job_executor.recreate_model_folder(distributed_job.extraction_identifier)
118118

0 commit comments

Comments
 (0)