From 82d443bc6e4a158cb59bce2219ddee11327c04fb Mon Sep 17 00:00:00 2001
From: Daniel King <daniel@mosaicml.com>
Date: Mon, 23 Sep 2024 14:54:51 -0700
Subject: [PATCH] pc

---
 tests/loggers/test_mosaic_ml_logger.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/tests/loggers/test_mosaic_ml_logger.py b/tests/loggers/test_mosaic_ml_logger.py
index 42c348029c..b4419d6f3a 100644
--- 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
\ No newline at end of file
+    assert mosaic_ml_logger.ignore_exceptions == True