Skip to content

Commit c80c7f3

Browse files
committed
fix up tests
1 parent 1804f7c commit c80c7f3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/lib/test_run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,7 @@ def test_run_with_file_output_iterator(self, respx_mock: MockRouter) -> None:
267267
return_value=httpx.Response(200, json=create_mock_prediction(output=file_urls))
268268
)
269269

270-
output: list[FileOutput] = self.client.run("some-model-ref", input={"prompt": "generate file iterator"})
270+
output: list[FileOutput] = self.client.run("some-model-ref", use_file_output=True, input={"prompt": "generate file iterator"})
271271

272272
assert isinstance(output, list)
273273
assert len(output) == 3
@@ -499,7 +499,7 @@ async def test_async_run_with_file_output_iterator(self, respx_mock: MockRouter)
499499
)
500500

501501
output: list[AsyncFileOutput] = await self.client.run(
502-
"some-model-ref", input={"prompt": "generate file iterator"}
502+
"some-model-ref", use_file_output=True, input={"prompt": "generate file iterator"}
503503
)
504504

505505
assert isinstance(output, list)

0 commit comments

Comments
 (0)