diff --git a/plugin/src/main/java/org/opensearch/ml/action/syncup/TransportSyncUpOnNodeAction.java b/plugin/src/main/java/org/opensearch/ml/action/syncup/TransportSyncUpOnNodeAction.java index 3afb2cff51..94327f190e 100644 --- a/plugin/src/main/java/org/opensearch/ml/action/syncup/TransportSyncUpOnNodeAction.java +++ b/plugin/src/main/java/org/opensearch/ml/action/syncup/TransportSyncUpOnNodeAction.java @@ -197,6 +197,9 @@ void cleanUpLocalCache(Map> runningDeployModelTasks) { return; } for (String taskId : allTaskIds) { + if (taskId == null) { + continue; + } MLTaskCache mlTaskCache = mlTaskManager.getMLTaskCache(taskId); // Task could be a prediction task, and it could be completed and removed from cache in predict thread during the cleaning up. if (mlTaskCache == null) {