added skipping to compressed-tensors

neuralmagic · Jul 1, 2024 · a699814 · a699814
1 parent 1cc7c46
commit a699814
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/models/test_compressed_tensors.py b/tests/models/test_compressed_tensors.py
@@ -9,8 +9,14 @@
 
 from tests.quantization.utils import is_quant_method_supported
 
+from tests.nm_utils.utils_skip import should_skip_test_group
+
 from .utils import check_logprobs_close
 
+if should_skip_test_group(group_name="TEST_MODELS"):
+    pytest.skip("TEST_MODELS=DISABLE, skipping model test group",
+                allow_module_level=True)
+
 MODELS = [
     "nm-testing/Meta-Llama-3-8B-Instruct-W8-Channel-A8-Dynamic-Per-Token-Test",
 ]