chore: fix FutureWarning test filtering

zama-ai · May 6, 2024 · c3f393f · c3f393f
1 parent 4f5be4f
commit c3f393f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/torch/test_hybrid_converter.py b/tests/torch/test_hybrid_converter.py
@@ -127,7 +127,6 @@ def test_gpt2_hybrid_mlp(list_or_str_private_modules_names, expected_accuracy):
     run_hybrid_llm_test(model, input_ids, list_or_str_private_modules_names, expected_accuracy)
 
 
-@pytest.mark.filterwarnings("ignore::FutureWarning")
 def test_hybrid_brevitas_qat_model():
     """Test GPT2 hybrid."""
     n_bits = 3
@@ -151,6 +150,7 @@ def test_hybrid_brevitas_qat_model():
 
 # Dependency 'huggingface-hub' raises a 'FutureWarning' from version 0.23.0 when calling the
 # 'from_pretrained' method
+@pytest.mark.filterwarnings("ignore::FutureWarning")
 def test_gpt2_hybrid_mlp_module_not_found():
     """Test GPT2 hybrid."""