Run ruff

allenai · Apr 24, 2024 · d99b8b8 · d99b8b8
1 parent 7226133
commit d99b8b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/hf_olmo/hf_olmo_test.py b/tests/hf_olmo/hf_olmo_test.py
@@ -9,7 +9,10 @@
 from olmo.torch_util import seed_all
 
 
-@pytest.mark.skipif(version.parse(transformers.__version__) >= version.parse("4.40.0"), reason="hf_olmo auto classes are not compatible with transformers >=v4.40.0")
+@pytest.mark.skipif(
+    version.parse(transformers.__version__) >= version.parse("4.40.0"),
+    reason="hf_olmo auto classes are not compatible with transformers >=v4.40.0",
+)
 def test_auto_hf_classes(model_path: str):
     from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer