From 87534309fe7dc8899edaca97cc4e7239800c9277 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Tue, 3 Oct 2023 09:47:07 -0700 Subject: [PATCH] precommit --- tests/test_hf_config.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_hf_config.py b/tests/test_hf_config.py index 1c7f255b35..5b3bb3d150 100644 --- 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',