From b9268548717ff56d347411cc8114c95233de2ae3 Mon Sep 17 00:00:00 2001 From: hengtaoguo Date: Thu, 5 Feb 2026 01:24:03 +0000 Subject: [PATCH] test --- tests/integration/generate_param_only_checkpoint_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/generate_param_only_checkpoint_test.py b/tests/integration/generate_param_only_checkpoint_test.py index fa96bab2c8..522f65eb3a 100644 --- a/tests/integration/generate_param_only_checkpoint_test.py +++ b/tests/integration/generate_param_only_checkpoint_test.py @@ -126,7 +126,7 @@ def test_param_ckpt_generation_with_dot_product(quantization, capsys): @pytest.mark.integration_test @pytest.mark.tpu_only -@pytest.mark.scheduled_only +# @pytest.mark.scheduled_only @pytest.mark.external_serving # Requires pre-generated checkpoint (Gemma-2b) def test_param_ckpt_generation_with_pre_generated_ckpt(capsys): """Tests the parameter-only checkpoint generation and decode flow with a pre-generated Gemma-2b model checkpoint."""