Update tests/test_litellm/llms/vertex_ai/test_vertex_ai_batch_transformation.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Sameer Kankute 2026-03-18 09:48:24 +05:30 committed by GitHub
parent 74ae17d153
commit d0d593beb8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -44,7 +44,6 @@ def test_output_file_id_falls_back_to_output_uri_prefix_with_predictions_jsonl()
)
@pytest.mark.asyncio
def test_vertex_ai_cancel_batch():
"""Test that vertex_ai cancel_batch calls the correct API endpoint"""
handler = VertexAIBatchPrediction(gcs_bucket_name="test-bucket")