added if clause to bypass the integration test

Signed-off-by: Dhrubo Saha <[email protected]>
opensearch-project · Oct 3, 2023 · 0bdf991 · 0bdf991
1 parent 9d17843
commit 0bdf991
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/src/main/java/org/opensearch/ml/task/MLPredictTaskRunner.java b/plugin/src/main/java/org/opensearch/ml/task/MLPredictTaskRunner.java
@@ -204,7 +204,9 @@ private void predict(String modelId, MLTask mlTask, MLInput mlInput, ActionListe
         mlStats
             .createCounterStatIfAbsent(mlTask.getFunctionName(), ActionName.PREDICT, MLActionLevelStat.ML_ACTION_REQUEST_COUNT)
             .increment();
-        mlStats.createModelCounterStatIfAbsent(modelId, ActionName.PREDICT, MLActionLevelStat.ML_ACTION_REQUEST_COUNT).increment();
+        if (modelId != null) {
+            mlStats.createModelCounterStatIfAbsent(modelId, ActionName.PREDICT, MLActionLevelStat.ML_ACTION_REQUEST_COUNT).increment();
+        }
         mlTask.setState(MLTaskState.RUNNING);
         mlTaskManager.add(mlTask);