From 4baa48ba4bd28467d241aa3d2b0352f38d57b09d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 15 Jul 2024 20:51:29 -0700 Subject: [PATCH] fix test_sync_embedding --- litellm/tests/test_custom_callback_input.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index 55edb8a79..c10a95b40 100644 --- a/litellm/tests/test_custom_callback_input.py +++ b/litellm/tests/test_custom_callback_input.py @@ -840,6 +840,7 @@ def test_sync_embedding(): ) print(f"customHandler_success.errors: {customHandler_success.errors}") print(f"customHandler_success.states: {customHandler_success.states}") + time.sleep(2) assert len(customHandler_success.errors) == 0 assert len(customHandler_success.states) == 3 # pre, post, success # test failure callback @@ -854,6 +855,7 @@ def test_sync_embedding(): pass print(f"customHandler_failure.errors: {customHandler_failure.errors}") print(f"customHandler_failure.states: {customHandler_failure.states}") + time.sleep(2) assert len(customHandler_failure.errors) == 1 assert len(customHandler_failure.states) == 3 # pre, post, failure except Exception as e: