diff --git a/tests/e2e/patched/test_llama_s2_attention.py b/tests/e2e/patched/test_llama_s2_attention.py index f1d37eb3c..010df9f99 100644 --- a/tests/e2e/patched/test_llama_s2_attention.py +++ b/tests/e2e/patched/test_llama_s2_attention.py @@ -7,6 +7,8 @@ import os import unittest from pathlib import Path +import pytest + from axolotl.cli import load_datasets from axolotl.common.cli import TrainerCliArgs from axolotl.train import train @@ -19,6 +21,7 @@ LOG = logging.getLogger("axolotl.tests.e2e") os.environ["WANDB_DISABLED"] = "true" +@pytest.mark.skip("Skipping test due to timeout.") class TestLlamaShiftedSparseAttention(unittest.TestCase): """ Test case for Llama models using S2 Attn