From e443d0192508a97509bb2af55c2430c485deede7 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 9 Sep 2025 20:37:09 -0700 Subject: [PATCH] test: remove redundant test --- tests/local_testing/test_router.py | 39 ++---------------------------- 1 file changed, 2 insertions(+), 37 deletions(-) diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index 007822a93b..e7e61d378c 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -1136,7 +1136,7 @@ async def test_aimg_gen_on_router(): "api_base": os.getenv("AZURE_SWEDEN_API_BASE"), "api_key": os.getenv("AZURE_SWEDEN_API_KEY"), }, - } + }, ] router = Router(model_list=model_list, num_retries=3) response = await router.aimage_generation( @@ -1308,41 +1308,6 @@ def test_azure_embedding_on_router(): # test_azure_embedding_on_router() -def test_bedrock_on_router(): - litellm.set_verbose = True - print("\n Testing bedrock on router\n") - try: - model_list = [ - { - "model_name": "claude-v1", - "litellm_params": { - "model": "bedrock/anthropic.claude-instant-v1", - }, - "tpm": 100000, - "rpm": 10000, - }, - ] - - async def test(): - router = Router(model_list=model_list) - response = await router.acompletion( - model="claude-v1", - messages=[ - { - "role": "user", - "content": "hello from litellm test", - } - ], - ) - print(response) - router.reset() - - asyncio.run(test()) - except Exception as e: - traceback.print_exc() - pytest.fail(f"Error occurred: {e}") - - # test_bedrock_on_router() @@ -2786,4 +2751,4 @@ def test_router_get_model_group_info(): assert model_group_info is not None assert model_group_info.model_group == "gpt-4" assert model_group_info.input_cost_per_token > 0 - assert model_group_info.output_cost_per_token > 0 \ No newline at end of file + assert model_group_info.output_cost_per_token > 0