Skip to content

Commit d78e250

Browse files
speedstorm1copybara-github
authored andcommitted
chore: Enable replay tests for Async Java batches module
PiperOrigin-RevId: 785614297
1 parent 69dad97 commit d78e250

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

google/genai/tests/batches/test_delete.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,10 @@
7777

7878
@pytest.mark.asyncio
7979
async def test_async_delete(client):
80-
with pytest_helper.exception_if_mldev(client, ValueError):
81-
delete_job = await client.aio.batches.delete(name=_BATCH_JOB_NAME)
80+
if client.vertexai:
81+
name = _BATCH_JOB_NAME
82+
else:
83+
name = _MLDEV_BATCH_OPERATION_NAME
8284

83-
assert delete_job
85+
delete_job = await client.aio.batches.delete(name=name)
86+
assert delete_job

0 commit comments

Comments
 (0)