Skip to content

Commit

Permalink
change XContentFactory to MediaTypeRegistry builder in MLRegisterMode…
Browse files Browse the repository at this point in the history
…lInputTest class

Signed-off-by: Bhavana Ramaram <rbhavna@amazon.com>
  • Loading branch information
rbhavna authored and ylwu-amzn committed Nov 20, 2023
1 parent b72cc90 commit 9f438d3
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -277,7 +277,7 @@ public void testMCorrInput() throws IOException {
.deployModel(true)
.modelNodeIds(new String[]{"modelNodeIds" })
.build();
XContentBuilder builder = XContentFactory.jsonBuilder();
XContentBuilder builder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
mcorrInput.toXContent(builder, ToXContent.EMPTY_PARAMS);
String jsonStr = builder.toString();
assertEquals(testString, jsonStr);
Expand Down

0 comments on commit 9f438d3

Please sign in to comment.