/ totaljobs

Author: b-d-e

src/kairos/logging.py

@@ -16,6 +16,7 @@ def __init__(self, log_dir: Path):
         """Initialise log directory."""
         self.log_dir = log_dir
         self.setup_main_logger()
+        self.total_jobs = 0
 
     def setup_main_logger(self):
         """Set up the main scheduler logger for scheduling events only."""
@@ -88,8 +89,8 @@ def log_job_start(
 
         # Log to scheduler log (minimal info)
         self.logger.info(
-            f"Starting job {job_index} ({job_info['job_name']}) on\
-                GPU {gpu_id} slot {slot_id}"
+            f"Starting job {job_index}/{self.total_jobs} "
+            f"({job_info['job_name']}) on GPU {gpu_id} slot {slot_id}"
         )
 
         # Log full details to job log
@@ -116,7 +117,8 @@ def log_job_completion(
 
         # Log to scheduler log (minimal info)
         self.logger.info(
-            f"Job {job_index} ({completion_info['job_name']}) completed "
+            f"Job {job_index}/{self.total_jobs} "
+            f"({completion_info['job_name']}) completed "
             f"with status: {completion_info['status']}"
         )
 

src/kairos/scheduler.py

@@ -142,6 +142,7 @@ def run_jobs(self, jobs: List[Job]) -> List[int]:
         job_queue = Queue()
         slot_queue = Queue()
         self.total_jobs = len(jobs)
+        self.job_logger.total_jobs = self.total_jobs
 
         for slot in self.gpu_slots:
             slot_queue.put(slot)
@@ -192,11 +193,3 @@ def worker():
             concurrent.futures.wait(futures)
 
             self.logger.info("\nAll jobs completed!")
-            self.logger.info("Results summary:")
-            for i, result in enumerate(results):
-                status = (
-                    "Success" if result == 0 else f"Failed (code {result})"
-                )
-                self.logger.info(f"Job {i}: {status}")
-
-            return results