From 4e70254d5611bf640a018d661eda057c6f4273fb Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Fri, 13 Mar 2026 12:32:55 -0700 Subject: [PATCH] [Fix] Increase _delete_file retry budget to reduce flakiness Increase max_retries from 6 to 9 and retry_delay from 10s to 20s (180s total wait, up from 60s) to give batch cost tracking more time to finish before cleanup attempts file deletion. Co-Authored-By: Claude Opus 4.6 --- .../test_proxy_e2e_azure_batches.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 cfdea73cefa..3a6ed5244c1 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 @@ -205,7 +205,7 @@ class TestManagedFilesAPI(ManagedFilesBase, UserKeyTestMixin): return metadata - def _delete_file(self, file_id, label, max_retries=6, retry_delay=10): + def _delete_file(self, file_id, label, max_retries=9, retry_delay=20): print(f"\nDeleting {label}: {self.shorten_id(file_id)}") for attempt in range(max_retries): try: