File tree Expand file tree Collapse file tree 2 files changed +3
-5
lines changed
main/java/org/elasticsearch/xpack/esql/inference
test/java/org/elasticsearch/xpack/esql/inference Expand file tree Collapse file tree 2 files changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -169,9 +169,7 @@ public Page buildOutput() {
169169
170170 @ Override
171171 public void close () {
172- inputPage .allowPassingToDifferentDriver ();
173- inputPage .releaseBlocks ();
174-
172+ releasePageOnAnyThread (inputPage );
175173 Releasables .closeExpectNoException (scoreBlockBuilder );
176174 }
177175
Original file line number Diff line number Diff line change @@ -71,7 +71,7 @@ private void initChannels() {
7171 scoreChannel = between (0 , channelCount - 1 );
7272 inputChannelElementTypes = IntStream .range (0 , channelCount ).sorted ().mapToObj (this ::randomElementType ).collect (Collectors .toList ());
7373 rowEncoderFactory = mockRowEncoderFactory ();
74- encodedRowLength = between (1 , 10 );
74+ encodedRowLength = between (1 , 20 );
7575 }
7676
7777 @ Before
@@ -236,7 +236,7 @@ private XContentRowEncoder.Factory mockRowEncoderFactory() {
236236 XContentRowEncoder rowEncoder = mock (XContentRowEncoder .class );
237237 doAnswer (encoderInvocation -> {
238238 Page inputPage = encoderInvocation .getArgument (0 , Page .class );
239- BytesRef scratch = new BytesRef (randomRealisticUnicodeOfCodepointLength (encodedRowLength ));
239+ BytesRef scratch = new BytesRef (randomAlphanumericOfLength (encodedRowLength ));
240240 return driverContext .blockFactory ().newConstantBytesRefBlockWith (scratch , inputPage .getPositionCount ());
241241 }).when (rowEncoder ).eval (any (Page .class ));
242242
You can’t perform that action at this time.
0 commit comments