review: add missing test guards

huggingface · Aug 22, 2024 · 3345bef · 3345bef
1 parent dca1a05
commit 3345bef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/test/models/test_quantized_model_for_causal_lm.py b/test/models/test_quantized_model_for_causal_lm.py
@@ -103,6 +103,7 @@ def test_quantized_model_for_causal_lm_sharded():
     compare_models(quantized, requantized)
 
 
+@pytest.mark.skipif(not is_transformers_available(), reason="requires transformers")
 @pytest.mark.parametrize("in_org", [True, False], ids=["org", "user"])
 def test_causal_lm_base_push_to_hub(staging, in_org):
     from optimum.quanto import QuantizedModelForCausalLM

diff --git a/test/models/test_quantized_model_for_pixart.py b/test/models/test_quantized_model_for_pixart.py
@@ -90,6 +90,7 @@ def test_quantized_model_for_pixart(qtype, exclude_proj_out):
     compare_models(quantized, requantized)
 
 
+@pytest.mark.skipif(not is_diffusers_available(), reason="requires diffusers")
 @pytest.mark.parametrize("in_org", [True, False], ids=["org", "user"])
 def test_push_to_hub(staging, in_org):
     from optimum.quanto import QuantizedPixArtTransformer2DModel