Clémentine commited on
Commit
71ff8d6
·
1 Parent(s): 5312397

fix duration

Browse files
Files changed (1) hide show
  1. utils/jobs.py +7 -0
utils/jobs.py CHANGED
@@ -271,6 +271,7 @@ def update_job_statuses() -> None:
271
 
272
  # Calculate aggregate statistics from completed runs
273
  completed_scores = [run["score"] for run in runs if run["status"] == "COMPLETED" and run["score"] is not None]
 
274
 
275
  if completed_scores:
276
  import statistics
@@ -282,6 +283,12 @@ def update_job_statuses() -> None:
282
 
283
  print(f"Aggregated {len(completed_scores)} runs: mean={mean_score:.4f}, variance={variance:.6f}")
284
 
 
 
 
 
 
 
285
  # Update completion time to latest run
286
  latest_completion = max([run["completed_at"] for run in runs if run.get("completed_at")], default=None)
287
  if latest_completion:
 
271
 
272
  # Calculate aggregate statistics from completed runs
273
  completed_scores = [run["score"] for run in runs if run["status"] == "COMPLETED" and run["score"] is not None]
274
+ completed_durations = [run["duration"] for run in runs if run["status"] == "COMPLETED" and run.get("duration") is not None]
275
 
276
  if completed_scores:
277
  import statistics
 
283
 
284
  print(f"Aggregated {len(completed_scores)} runs: mean={mean_score:.4f}, variance={variance:.6f}")
285
 
286
+ # Calculate average duration
287
+ if completed_durations:
288
+ mean_duration = statistics.mean(completed_durations)
289
+ globals.job_results[key]["duration"] = mean_duration
290
+ print(f"Average duration: {mean_duration:.2f} seconds")
291
+
292
  # Update completion time to latest run
293
  latest_completion = max([run["completed_at"] for run in runs if run.get("completed_at")], default=None)
294
  if latest_completion: