Skip to content

Commit 4e9e129

Browse files
committed
debugging
Signed-off-by: Sicheng Song <sicheng.song@outlook.com>
1 parent fc902af commit 4e9e129

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

plugin/src/main/java/org/opensearch/ml/action/prediction/TransportPredictionTaskAction.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,6 @@ public void onResponse(MLModel mlModel) {
156156
)
157157
);
158158
} else {
159-
log.info("ModelId " + modelId + " is enabled for prediction in local layer.");
160-
log.info(mlPredictionTaskRequest.getMlInput().toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
161159
validateInputSchema(modelId, mlPredictionTaskRequest.getMlInput());
162160
executePredict(mlPredictionTaskRequest, wrappedListener, modelId);
163161
}
@@ -240,6 +238,8 @@ private void executePredict(
240238

241239
private void validateInputSchema(String modelId, MLInput mlInput) {
242240
log.info("Went to validate input schema");
241+
log.info(modelCacheHelper.getModelInterface(modelId).containsKey("input"));
242+
log.info(modelCacheHelper.getModelInterface(modelId).get("input"));
243243
if (modelCacheHelper.getModelInterface(modelId) != null && modelCacheHelper.getModelInterface(modelId).get("input") != null) {
244244
String inputSchemaString = modelCacheHelper.getModelInterface(modelId).get("input");
245245
log.info(inputSchemaString);

0 commit comments

Comments
 (0)