diff --git a/src/raythena/actors/esworker.py b/src/raythena/actors/esworker.py index 0156efa..f13e5f6 100644 --- a/src/raythena/actors/esworker.py +++ b/src/raythena/actors/esworker.py @@ -76,6 +76,7 @@ DONE: [READY_FOR_JOB], } + @ray.remote(num_cpus=1, max_restarts=1, max_task_retries=3) class ESWorker: """ @@ -323,9 +324,7 @@ def transition_state(self, dest: int) -> None: IllegalWorkerState if the transition isn't allowed """ if dest not in self.transitions[self.state]: - self._logger.error( - f"Illegal transition from {STATES_NAME[self.state]} to {STATES_NAME[dest]}" - ) + self._logger.error(f"Illegal transition from {STATES_NAME[self.state]} to {STATES_NAME[dest]}") raise IllegalWorkerState( worker_id=self.id, src_state=STATES_NAME[self.state], @@ -576,9 +575,7 @@ def get_message(self) -> WorkerResponse: self.transition_state(STAGE_OUT) self.stageout() return self.return_message(Messages.PROCESS_DONE) - elif self.is_event_service_job() and ( - self.state == READY_FOR_EVENTS or self.should_request_ranges() - ): + elif self.is_event_service_job() and (self.state == READY_FOR_EVENTS or self.should_request_ranges()): req = EventRangeRequest() req.add_event_request( self.job["PandaID"], diff --git a/src/raythena/utils/config.py b/src/raythena/utils/config.py index 57b70cf..6a95457 100644 --- a/src/raythena/utils/config.py +++ b/src/raythena/utils/config.py @@ -41,6 +41,7 @@ }, } + class Config: """Class storing app configuration. diff --git a/src/raythena/utils/eventservice.py b/src/raythena/utils/eventservice.py index 5c89904..4e14c5f 100644 --- a/src/raythena/utils/eventservice.py +++ b/src/raythena/utils/eventservice.py @@ -1073,7 +1073,7 @@ class EventRange: FAILED = "failed" FATAL = "fatal" STATES = frozenset([READY, ASSIGNED, DONE, FAILED, FATAL]) - + def __init__( self, event_range_id: str, diff --git a/src/raythena/utils/exception.py b/src/raythena/utils/exception.py index aca2ed4..eff9a0c 100644 --- a/src/raythena/utils/exception.py +++ b/src/raythena/utils/exception.py @@ -9,12 +9,13 @@ UNKNOWN = 0 ERROR_CODES_GENRIC_MESSAGES = { - ILLEGAL_WORKER_STATE: "Illegal worker state transition", - STAGEIN_FAILED: "Failed to stagein data", - STAGEOUT_FAILED: "Failed to stageout data", - PAYLOAD_FAILED: "Payload execution failed", - UNKNOWN: "Unknown error", - } + ILLEGAL_WORKER_STATE: "Illegal worker state transition", + STAGEIN_FAILED: "Failed to stagein data", + STAGEOUT_FAILED: "Failed to stageout data", + PAYLOAD_FAILED: "Payload execution failed", + UNKNOWN: "Unknown error", +} + class ErrorCodes: """