pyright

EleutherAI · Jul 20, 2023 · 785537b · 785537b
1 parent 0d6c8b9
commit 785537b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elk/training/multi_reporter.py b/elk/training/multi_reporter.py
@@ -29,8 +29,8 @@ def __init__(self, reporter: list[ReporterWithInfo]):
         )
 
         self.train_loss = (
-            sum(train_losses) / len(train_losses)
-            if train_losses is not None  # type: ignore
+            sum(train_losses) / len(train_losses)  # type: ignore
+            if train_losses is not None
             else None
         )