Return status

ITISFoundation · Jul 10, 2024 · a1b498a · a1b498a
1 parent 98d788f
commit a1b498a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docker_scripts/parallelrunner.py b/docker_scripts/parallelrunner.py
@@ -326,14 +326,15 @@ def run_job(self, job_inputs):
 
         return job_outputs, status
 
-    def process_job_outputs(self, results, batch):
+    def process_job_outputs(self, results, batch, status):
         if self.template_id == "TEST_UUID":
             logger.info("Map in test mode, just returning input")
 
             return batch
 
         for task_i, task in enumerate(batch):
             output = task["output"]
+            task["status"] = status
             for probe_name, probe_outputs in results.items():
                 if probe_name not in output:
                     raise ValueError(f"Unknown probe in output: {probe_name}")
@@ -433,7 +434,7 @@ def map_func(batch, trial_number=1):
 
                 job_outputs, status = self.run_job(job_inputs)
 
-                batch = self.process_job_outputs(job_outputs, batch)
+                batch = self.process_job_outputs(job_outputs, batch, status)
 
                 logger.info(
                     "Worker has finished batch "