fix: Adding code identities (#185)

AstraZeneca · Jan 14, 2025 · 57aeb48 · 57aeb48
1 parent cea2470
commit 57aeb48
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/extensions/job_executor/k8s.py b/extensions/job_executor/k8s.py
@@ -190,6 +190,7 @@ def execute_job(self, job: BaseTaskType, catalog_settings=Optional[List[str]]):
         self._use_volumes()
 
         job_log = self._context.run_log_store.get_job_log(run_id=self._context.run_id)
+        self.add_code_identities(job_log)
 
         attempt_log = job.execute_command(
             attempt_number=self.step_attempt_number,
@@ -428,6 +429,7 @@ def execute_job(self, job: BaseTaskType, catalog_settings=Optional[List[str]]):
         )
 
         job_log = self._context.run_log_store.get_job_log(run_id=self._context.run_id)
+        self.add_code_identities(job_log)
 
         attempt_log = job.execute_command(
             attempt_number=self.step_attempt_number,

diff --git a/extensions/job_executor/local.py b/extensions/job_executor/local.py
@@ -37,6 +37,7 @@ def execute_job(self, job: BaseTaskType, catalog_settings=Optional[List[str]]):
         logger.info("Trying to execute job")
 
         job_log = self._context.run_log_store.get_job_log(run_id=self._context.run_id)
+        self.add_code_identities(job_log)
 
         attempt_log = job.execute_command(
             attempt_number=self.step_attempt_number,

diff --git a/extensions/job_executor/local_container.py b/extensions/job_executor/local_container.py
@@ -52,6 +52,7 @@ def execute_job(self, job: BaseTaskType, catalog_settings=Optional[List[str]]):
         logger.info("Trying to execute job")
 
         job_log = self._context.run_log_store.get_job_log(run_id=self._context.run_id)
+        self.add_code_identities(job_log)
 
         attempt_log = job.execute_command(
             attempt_number=self.step_attempt_number,