precommit

mosaicml · Oct 3, 2023 · 8753430 · 8753430
1 parent 575cb13
commit 8753430
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_hf_config.py b/tests/test_hf_config.py
@@ -1,9 +1,9 @@
 # Copyright 2022 MosaicML LLM Foundry authors
 # SPDX-License-Identifier: Apache-2.0
 
+import os
 import tempfile
 from copy import deepcopy
-import os
 from pathlib import Path
 from typing import Any, Dict, Mapping
 
@@ -142,7 +142,8 @@ def test_hf_config_override(
             assert getattr(hf_model.config, k) == v
 
 
-@pytest.mark.skipif('HUGGING_FACE_HUB_TOKEN' not in os.environ, reason='CI does not have access to llama2')
+@pytest.mark.skipif('HUGGING_FACE_HUB_TOKEN' not in os.environ,
+                    reason='CI does not have access to llama2')
 def test_rope_scaling_override():
     model_cfg = {
         'name': 'hf_causal_lm',