TST: Skiping AWQ tests for now .. (#1690)

* Update test_gpu_examples.py * Update tests/test_gpu_examples.py
huggingface · Apr 29, 2024 · 608a90d · 608a90d
1 parent e19f7bf
commit 608a90d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_gpu_examples.py b/tests/test_gpu_examples.py
@@ -1930,8 +1930,10 @@ def test_causal_lm_training_aqlm(self):
             assert trainer.state.log_history[-1]["train_loss"] is not None
 
 
+# TODO: unskip the tests once https://github.com/casper-hansen/AutoAWQ/issues/466 is fixed
 @require_torch_gpu
 @require_auto_awq
+@pytest.mark.skip(reason="Needs https://github.com/casper-hansen/AutoAWQ/issues/466 to be fixed first")
 class PeftAwqGPUTests(unittest.TestCase):
     r"""
     Awq + peft tests