From a0600a30d83ae48c7391be95ac432abc0957d4a9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 22 Jul 2024 15:32:49 -0700 Subject: [PATCH] fix using arize as success callback --- litellm/tests/test_arize_ai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_arize_ai.py b/litellm/tests/test_arize_ai.py index 7c38db4c6..dfc00446e 100644 --- a/litellm/tests/test_arize_ai.py +++ b/litellm/tests/test_arize_ai.py @@ -18,7 +18,7 @@ import logging @pytest.mark.asyncio() async def test_async_otel_callback(): litellm.set_verbose = True - litellm.callbacks = ["arize"] + litellm.success_callback = ["arize"] await litellm.acompletion( model="gpt-3.5-turbo",