re-raise

OasisLMF · Jul 2, 2024 · f912571 · f912571
1 parent a04d89b
commit f912571
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/model_execution_worker/distributed_tasks.py b/src/model_execution_worker/distributed_tasks.py
@@ -448,6 +448,7 @@ def run(self, params, *args, run_data_uuid=None, analysis_id=None, **kwargs):
             except Exception as error:
                 # fallback only needed if celery can't serialize the exception
                 logger.exception("Error occured in 'keys_generation_task':")
+                raise error
 
     return run
 
@@ -863,6 +864,7 @@ def run(self, params, *args, run_data_uuid=None, analysis_id=None, **kwargs):
             except Exception as error:
                 # fallback only needed if celery can't serialize the exception
                 logger.exception("Error occured in 'loss_generation_task':")
+                raise error
 
     return run