From c3f393f5f672cdeae1645530b5a4f9bf4517ddfb Mon Sep 17 00:00:00 2001 From: Roman Bredehoft Date: Mon, 6 May 2024 10:28:33 +0200 Subject: [PATCH] chore: fix FutureWarning test filtering --- tests/torch/test_hybrid_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/torch/test_hybrid_converter.py b/tests/torch/test_hybrid_converter.py index 64acffec3..b2ce6e07c 100644 --- 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."""