Updated the tests to account for the epoch arriving at the logger.

Lightning-AI · Apr 23, 2020 · d5194f2 · d5194f2
1 parent e69f20d
commit d5194f2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
@@ -62,9 +62,9 @@ def log_metrics(self, metrics, step):
     trainer.test()
 
     log_metric_names = [(s, sorted(m.keys())) for s, m in logger.history]
-    assert log_metric_names == [(0, ['val_acc', 'val_loss']),
-                                (0, ['train_some_val']),
-                                (1, ['test_acc', 'test_loss'])]
+    assert log_metric_names == [(0, ['epoch', 'val_acc', 'val_loss']),
+                                (0, ['epoch', 'train_some_val']),
+                                (1, ['epoch', 'test_acc', 'test_loss'])]
 
 
 @pytest.mark.parametrize("logger_class", [