Update tests/loggers/test_base.py

Lightning-AI · Sep 10, 2021 · df63517 · df63517
1 parent 6d12d69
commit df63517
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/loggers/test_base.py b/tests/loggers/test_base.py
@@ -48,7 +48,7 @@ def test_logger_collection():
     mock1.agg_and_log_metrics.assert_called_once_with({"test": 2.0}, 4)
     mock2.agg_and_log_metrics.assert_called_once_with({"test": 2.0}, 4)
 
-    logger.finalize()
+    logger.finalize("success")
     mock1.finalize.assert_called_once()
     mock2.finalize.assert_called_once()