From 94f7751c83cd123ef7e545d12f06f8acca93ef42 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 9 Dec 2024 11:39:59 -0500 Subject: [PATCH] remove duplicated functionality fixture --- tests/e2e/patched/test_fa_xentropy.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/tests/e2e/patched/test_fa_xentropy.py b/tests/e2e/patched/test_fa_xentropy.py index effcb39c7d..183843b7b1 100644 --- 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