move decorator to test method instead of class
This commit is contained in:
@@ -30,12 +30,12 @@ MODELS = [
|
||||
@pytest.mark.parametrize(
|
||||
"save_compressed", [True, False], ids=["save_compressed", "save_uncompressed"]
|
||||
)
|
||||
@require_llmcompressor
|
||||
class TestLLMCompressorIntegration:
|
||||
"""
|
||||
e2e tests for axolotl.integrations.llm_compressor.LLMCompressorPlugin
|
||||
"""
|
||||
|
||||
@require_llmcompressor
|
||||
@require_torch_2_4_1
|
||||
def test_llmcompressor_plugin(
|
||||
self, temp_dir, base_model: str, save_compressed: bool
|
||||
|
||||
@@ -105,7 +105,7 @@ def require_vllm(test_case):
|
||||
return False
|
||||
|
||||
return unittest.skipUnless(
|
||||
is_vllm_installed(), "test requires a vllm to be installed"
|
||||
is_vllm_installed(), "test requires vllm to be installed"
|
||||
)(test_case)
|
||||
|
||||
|
||||
@@ -123,7 +123,7 @@ def require_llmcompressor(test_case):
|
||||
return False
|
||||
|
||||
return unittest.skipUnless(
|
||||
is_llmcompressor_installed(), "test requires a llmcompressor to be installed"
|
||||
is_llmcompressor_installed(), "test requires llmcompressor to be installed"
|
||||
)(test_case)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user