Update app.py

app.py

@@ -115,7 +115,7 @@ def fetch_wandb_data():
 
  submissions: set[int] = set()
  benchmarks: set[int] = set()
- invalid: list[int] = []
+ invalid: dict[int, str] = {}
  average_benchmark_time = 0.0
 
  for key, value in run.summary.items():
@@ -123,7 +123,11 @@ def fetch_wandb_data():
  average_benchmark_time = float(value)
 
  elif key == "invalid":
- invalid = value
+ try:
+ invalid = dict(value)
+ except TypeError:
+ invalid = {uid: "Unknown" for uid in value}
+
 
  elif key == "submissions":
  submissions = {int(uid) for uid in value.keys()}
@@ -131,7 +135,7 @@ def fetch_wandb_data():
  elif key == "benchmarks":
  benchmarks = {int(uid) for uid in value.keys()}
 
- benchmarks = set(benchmarks) - set(invalid)
+ benchmarks = set(benchmarks) - set(invalid.keys())
 
  status = Status.BENCHMARKING
  run_state = run.state