Fixed response map

Signed-off-by: Sahil Buddharaju <sahilbud@amazon.com>
buddharajusahil · Dec 10, 2024 · 84f8736 · 84f8736
1 parent f546760
commit 84f8736
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/test/java/org/opensearch/knn/plugin/action/RestTrainModelHandlerIT.java b/src/test/java/org/opensearch/knn/plugin/action/RestTrainModelHandlerIT.java
@@ -545,7 +545,7 @@ public void testTrainModel_success_methodOverrideWithCompressionMode() throws Ex
         String responseBody = EntityUtils.toString(getResponse.getEntity());
         assertNotNull(responseBody);
 
-        Map<String, Object> responseMap = createParser(MediaTypeRegistry.getDefaultMediaType().xContent(), responseBody).map();
+        Map<String, Object> responseMap = createParser(XContentType.JSON.xContent(), responseBody).map();
 
         assertEquals(modelId, responseMap.get(MODEL_ID));