Check if initializing

app.py

@@ -26,11 +26,12 @@ demo = gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}",
 
 
 class Status(Enum):
- IN_PROGRESS = ("In Progress", "orange")
+ BENCHMARKING = ("Benchmarking", "orange")
  DONE = ("Done", "springgreen")
  BROKEN = ("Broken", "red")
  STOPPED = ("Stopped", "red")
  CRASHED = ("Crashed", "red")
+ INITIALIZING = ("Initializing", "orange")
 
  def get_alt_time_text(self) -> str:
  return "∞" if self.is_broken() else "N/A"
@@ -42,7 +43,7 @@ class Status(Enum):
  return self.value[1]
 
  def is_broken(self):
- return self == Status.BROKEN or self == Status.STOPPED or self == Status.CRASHED
+ return self == Status.BROKEN or self == Status.STOPPED or self == Status.CRASHED or self == Status.INITIALIZING
 
 
 @dataclass
@@ -140,7 +141,7 @@ def fetch_wandb_data():
  for benchmark_key, benchmark_value in value.items():
  benchmarks.add(int(benchmark_key))
 
- status = Status.IN_PROGRESS
+ status = Status.BENCHMARKING
  run_state = run.state
  if run_state == "finished":
  status = Status.STOPPED
@@ -149,7 +150,10 @@ def fetch_wandb_data():
  elif completed:
  status = Status.DONE
  elif not submissions or (not average_benchmark_time and benchmarks):
- status = Status.BROKEN
+ if 12 <= now.hour < 13:
+ status = Status.INITIALIZING
+ else:
+ status = Status.BROKEN
 
  block = subtensor.get_current_block()
  data[uid] = State(
@@ -200,7 +204,7 @@ def get_data() -> gr.Dataframe:
  eta = int(state.average_benchmark_time * (state.submissions - (state.benchmarks + state.invalid)))
  time_left = timedelta(seconds=eta)
  eta_date = now + time_left
- eta_time = eta_date.strftime("%Y-%m-%d %I:%M:%S %p") if eta > 0 and state.status == Status.IN_PROGRESS else state.status.get_alt_time_text()
+ eta_time = eta_date.strftime("%Y-%m-%d %I:%M:%S %p") if eta > 0 and state.status == Status.BENCHMARKING else state.status.get_alt_time_text()
 
  average_time_text = f"{timedelta(seconds=int(state.average_benchmark_time))}" if state.average_benchmark_time else state.status.get_alt_time_text(),
 
@@ -215,7 +219,7 @@ def get_data() -> gr.Dataframe:
  state.invalid,
  f"<span style='color: {'orange' if state.average_benchmark_time > AVERAGE_BENCHMARK_TIME_WARNING_THRESHOLD else 'springgreen'}'>{average_time_text[0]}</span>",
  f"<span style='color: {'orange' if eta > ETA_WARNING_THRESHOLD else 'springgreen'}'>{eta_time}</span>",
- f"<span style='color: {'orange' if eta > ETA_WARNING_THRESHOLD else 'springgreen'}'>{time_left if eta > 0 and state.status == Status.IN_PROGRESS else state.status.get_alt_time_text()}</span>",
+ f"<span style='color: {'orange' if eta > ETA_WARNING_THRESHOLD else 'springgreen'}'>{time_left if eta > 0 and state.status == Status.BENCHMARKING else state.status.get_alt_time_text()}</span>",
  f"<span style='color: {'springgreen' if state.vtrust > 0.75 else 'red'}'>{state.vtrust:.4f}</span>",
  f"<span style='color: {'springgreen' if state.updated < 1000 else 'red'}'>{state.updated}</span>",
  ))