Skip to content

Commit cb4f412

Browse files
committed
try and fix global logging
1 parent 6080600 commit cb4f412

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/kairos/scheduler.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,10 @@ def run_job(self, gpu_slot: GPUSlot, job: Job, job_index: int) -> int:
8080
job_logger,
8181
)
8282

83+
self.job_logger.log_job_start(
84+
job_logger, job, job_index, gpu_slot.gpu_id, gpu_slot.slot_id
85+
)
86+
8387
env = os.environ.copy()
8488
if job.env:
8589
env.update(job.env)
@@ -121,6 +125,9 @@ def run_job(self, gpu_slot: GPUSlot, job: Job, job_index: int) -> int:
121125
job_logger.info(
122126
f"Job completed: {json.dumps(completion_info, indent=2)}"
123127
)
128+
self.job_logger.log_job_completion(
129+
job_logger, job, job_index, gpu_slot.gpu_id, return_code
130+
)
124131

125132
return return_code
126133

0 commit comments

Comments
 (0)