diff --git a/src/omotes_sdk/internal/worker/worker.py b/src/omotes_sdk/internal/worker/worker.py index 2f4bf51..8ad5b44 100644 --- a/src/omotes_sdk/internal/worker/worker.py +++ b/src/omotes_sdk/internal/worker/worker.py @@ -22,16 +22,6 @@ logger = logging.getLogger("omotes_sdk_internal") -class EarlySystemExit(Exception): - """Wrapper for `SystemExit` exception. - - To ensure that the worker process does not shutdown but rather handles the `SystemExit` as an - error - """ - - ... - - class TaskUtil: """Utilities for a Celery task.""" @@ -217,10 +207,7 @@ def wrapped_worker_task( logger.info("Worker started new task %s", job_id) task_util = TaskUtil(job_id, task, task.broker_if) task_util.update_progress(0, "Job calculation started") - try: - task.output_esdl = WORKER_TASK_FUNCTION(input_esdl, params_dict, task_util.update_progress) - except SystemExit as e: - raise EarlySystemExit(e) + task.output_esdl = WORKER_TASK_FUNCTION(input_esdl, params_dict, task_util.update_progress) task_util.update_progress(1.0, "Calculation finished.")