add assert

elk/training/multi_reporter.py

@@ -18,6 +18,7 @@
 
 class MultiReporter:
  def __init__(self, reporter_results: list[ReporterTrainResult]):
+ assert len(reporter_results) > 0, "Must have at least one reporter"
  self.reporter_results: list[ReporterTrainResult] = reporter_results
  self.reporters = [r.reporter for r in reporter_results]
  train_losses = (
@@ -26,7 +27,7 @@
  else None
  )
  self.train_loss = (
  sum(train_losses) / len(train_losses) if train_losses is not None else None
  )
 
  def __call__(self, h):
@@ -46,4 +47,4 @@
  reporter = t.load(path, map_location=device)
  reporters.append(reporter)
  # TODO for now I don't care about the train losses
  return MultiReporter([ReporterTrainResult(r, None) for r in reporters])