[Fix] Add flaky reruns to test_e2e_managed_batch

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 <noreply@anthropic.com>
This commit is contained in:
yuneng-jiang 2026-03-13 11:57:00 -07:00
parent e40ebf262a
commit 1f71578de2

View file

@ -235,6 +235,7 @@ class TestManagedFilesAPI(ManagedFilesBase, UserKeyTestMixin):
# Tests
# ------------------------------------------------------------------
@pytest.mark.flaky(reruns=5)
@pytest.mark.parametrize(
"model_name",
get_batch_model_names(),