diff --git a/tests/openai_misc_endpoints_tests/test_openai_batches_endpoint.py b/tests/openai_misc_endpoints_tests/test_openai_batches_endpoint.py index 84f09c34cf4..d5170d5b242 100644 --- a/tests/openai_misc_endpoints_tests/test_openai_batches_endpoint.py +++ b/tests/openai_misc_endpoints_tests/test_openai_batches_endpoint.py @@ -162,8 +162,8 @@ def test_e2e_batches_files(custom_llm_provider): ) if custom_llm_provider == "azure": - # azure takes very long to complete a batch - randomly pick a completed batch - batch_id = get_any_completed_batch_id_azure() + # azure takes very long to complete a batch + return else: await_batch_completion( batch_id=batch_id, custom_llm_provider=custom_llm_provider