From bc7e3a4af481847c19f579f59619aafdf6248442 Mon Sep 17 00:00:00 2001 From: Shobhit Singh Date: Thu, 12 Sep 2024 23:17:00 +0000 Subject: [PATCH] test: disable claude3 predict tests to protect bqml capacity --- tests/system/small/ml/test_llm.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/system/small/ml/test_llm.py b/tests/system/small/ml/test_llm.py index e3d2b51081..cd5d65c458 100644 --- a/tests/system/small/ml/test_llm.py +++ b/tests/system/small/ml/test_llm.py @@ -439,6 +439,7 @@ def test_claude3_text_generator_create_load( assert reloaded_model.model_name == model_name +@pytest.mark.skip("b/366290533 too many requests are exhausting bqml capacity") @pytest.mark.parametrize( "model_name", ("claude-3-sonnet", "claude-3-haiku", "claude-3-5-sonnet", "claude-3-opus"), @@ -458,6 +459,7 @@ def test_claude3_text_generator_predict_default_params_success( ) +@pytest.mark.skip("b/366290533 too many requests are exhausting bqml capacity") @pytest.mark.parametrize( "model_name", ("claude-3-sonnet", "claude-3-haiku", "claude-3-5-sonnet", "claude-3-opus"),