pc

mosaicml · Sep 23, 2024 · 82d443b · 82d443b
1 parent 623772d
commit 82d443b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tests/loggers/test_mosaic_ml_logger.py b/tests/loggers/test_mosaic_ml_logger.py
@@ -1,9 +1,15 @@
-from llmfoundry.utils.builders import build_logger
+# Copyright 2024 MosaicML LLM Foundry authors
+# SPDX-License-Identifier: Apache-2.0
 
 from composer.loggers import MosaicMLLogger
 
+from llmfoundry.utils.builders import build_logger
+
+
 def test_mosaic_ml_logger_constructs():
-    mosaic_ml_logger = build_logger('mosaicml', kwargs={'ignore_exceptions': True})
+    mosaic_ml_logger = build_logger(
+        'mosaicml', kwargs={'ignore_exceptions': True}
+    )
 
     assert isinstance(mosaic_ml_logger, MosaicMLLogger)
-    assert mosaic_ml_logger.ignore_exceptions == True
+    assert mosaic_ml_logger.ignore_exceptions == True