style check

allenai · Dec 9, 2023 · 68ff059 · 68ff059
1 parent 64118d6
commit 68ff059
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/hf_olmo/hf_olmo_test.py b/tests/hf_olmo/hf_olmo_test.py
@@ -11,7 +11,7 @@ def test_auto_hf_classes(model_path: str):
     from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
 
     from hf_olmo import OLMoConfig, OLMoForCausalLM, OLMoTokenizerFast
-    from hf_olmo.convert_olmo_to_hf import write_config, write_tokenizer, write_model
+    from hf_olmo.convert_olmo_to_hf import write_config, write_model, write_tokenizer
 
     # model_path is an OLMo checkpoint.
     # Creates HF-compatible config.json