diff --git a/litellm/tests/test_batch_completions.py b/litellm/tests/test_batch_completions.py index 485f51757b9..cd5eb251e85 100644 --- a/litellm/tests/test_batch_completions.py +++ b/litellm/tests/test_batch_completions.py @@ -74,6 +74,8 @@ def test_batch_completion_models_all_responses(): assert len(responses) == 2 except Timeout as e: pass + except litellm.APIError as e: + pass except Exception as e: pytest.fail(f"An error occurred: {e}")