remove duplicated functionality fixture

axolotl-ai-cloud · Dec 9, 2024 · 94f7751 · 94f7751
1 parent 70099c5
commit 94f7751
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/tests/e2e/patched/test_fa_xentropy.py b/tests/e2e/patched/test_fa_xentropy.py
@@ -4,7 +4,6 @@
 
 import logging
 import os
-from importlib import reload
 from pathlib import Path
 
 import pytest
@@ -22,14 +21,6 @@
 os.environ["WANDB_DISABLED"] = "true"
 
 
-@pytest.fixture(autouse=True)
-def reload_transformers():
-    import transformers.models.llama.modeling_llama
-
-    yield
-    reload(transformers.models.llama.modeling_llama)
-
-
 class TestFAXentropyLlama:
     """
     Test case for Llama models using LoRA w multipack