Remove wainitng status

Author: gabriel-piles

src/trainable_entity_extractor/domain/DistributedSubJob.py

@@ -10,5 +10,5 @@ class DistributedSubJob(BaseModel):
     extractor_job: TrainableEntityExtractorJob
     retry_count: int = 0
     max_retries: int = 2
-    status: JobStatus = JobStatus.WAITING
+    status: JobStatus = JobStatus.PENDING
     result: Performance | bool | None = None

src/trainable_entity_extractor/domain/JobStatus.py

@@ -2,10 +2,8 @@
 
 
 class JobStatus(StrEnum):
-    WAITING = "WAITING"
     PENDING = "PENDING"
     RUNNING = "RUNNING"
     SUCCESS = "SUCCESS"
     FAILURE = "FAILURE"
     CANCELED = "CANCELED"
-    RETRY = "RETRY"

src/trainable_entity_extractor/use_cases/OrchestratorUseCase.py

@@ -50,7 +50,7 @@ def _process_training_job(self, distributed_job: DistributedJob) -> JobProcessin
         extraction_identifier = distributed_job.extraction_identifier
         sub_job = distributed_job.sub_jobs[0]
 
-        if sub_job.status == JobStatus.WAITING:
+        if sub_job.status == JobStatus.PENDING:
             self.job_executor.start_training(extraction_identifier, sub_job)
 
         if sub_job.status == JobStatus.SUCCESS:
@@ -132,7 +132,7 @@ def _start_pending_performance_evaluations(self, distributed_job: DistributedJob
                 self._cancel_and_remove_job(distributed_job)
                 return
 
-            if sub_job.status == JobStatus.WAITING:
+            if sub_job.status == JobStatus.PENDING:
                 self.job_executor.start_performance_evaluation(distributed_job.extraction_identifier, sub_job)
 
             if sub_job.result and hasattr(sub_job.result, "is_perfect") and sub_job.result.is_perfect: