fixed eta

src/wandb_data.py

@@ -209,6 +209,19 @@ def _add_runs(wandb_runs: list[wapi.Run]):
         date = _date_from_run(wandb_run)
         id = wandb_run.id
         average_benchmarking_time = float(wandb_run.summary["average_benchmarking_time"]) if "average_benchmarking_time" in wandb_run.summary else 0
+
+        # Get num_gpus from metrics, default to 1 if not found
+        num_gpus = int(metrics.get("num_gpus", 1))
+        
+        # Update ETA calculation to account for GPUs
+        eta_calculation = (
+            max(
+                int(average_benchmarking_time * (len(submission_info) - len(submissions) - len(invalid_submissions))) if average_benchmarking_time else 0, 
+                0
+            ) // num_gpus  # Divide by number of GPUs
+            if status != BenchmarkStatus.FINISHED else 0
+        )
+
         run = Run(
             start_date=date,
             version=version,
@@ -218,7 +231,7 @@ def _add_runs(wandb_runs: list[wapi.Run]):
             status=status,
             average_benchmarking_time=average_benchmarking_time,
             step=int(metrics["step"]),
-            eta=max(int(average_benchmarking_time * (len(submission_info) - len(submissions) - len(invalid_submissions))) if average_benchmarking_time else 0, 0) if status != BenchmarkStatus.FINISHED else 0,
+            eta=eta_calculation,
             winner_uid=winner_uid,
             baseline_metrics=baseline_metrics,
             total_submissions=len(submission_info),