Skip to content

Commit

Permalink
[ML] Always send at least 1 input (#113456)
Browse files Browse the repository at this point in the history
  • Loading branch information
prwhelan authored Sep 24, 2024
1 parent 6e24e5a commit 806a72a
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -260,7 +260,7 @@ public void testSupportedStream() throws Exception {
assertEquals(modelId, singleModel.get("inference_id"));
assertEquals(TaskType.COMPLETION.toString(), singleModel.get("task_type"));

var input = IntStream.range(0, randomInt(10)).mapToObj(i -> randomAlphaOfLength(10)).toList();
var input = IntStream.range(1, randomInt(10)).mapToObj(i -> randomAlphaOfLength(10)).toList();

try {
var events = streamInferOnMockService(modelId, TaskType.COMPLETION, input);
Expand Down

0 comments on commit 806a72a

Please sign in to comment.