From 682b2783d0820c73aa4d099432bc5be73ac15250 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 28 Dec 2024 19:55:49 -0800 Subject: [PATCH] test_e2e_batches_files --- .../test_openai_batches_endpoint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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