From 1f71578de2958b659dc640bc1b208064f2c1af6a Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Fri, 13 Mar 2026 11:57:00 -0700 Subject: [PATCH] [Fix] Add flaky reruns to test_e2e_managed_batch MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The test_e2e_managed_batch test intermittently fails during cleanup when deleting the input file — the batch cost tracking hasn't finished processing yet (batch_processed=true not set), causing a 400 error. This is a timing race condition unrelated to batch retrieval logic. Co-Authored-By: Claude Opus 4.6 --- .../test_proxy_e2e_azure_batches.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py b/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py index 262c55efc5d..cfdea73cefa 100644 --- a/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py +++ b/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py @@ -235,6 +235,7 @@ class TestManagedFilesAPI(ManagedFilesBase, UserKeyTestMixin): # Tests # ------------------------------------------------------------------ + @pytest.mark.flaky(reruns=5) @pytest.mark.parametrize( "model_name", get_batch_model_names(),