Fix null taskId causing model undeploy issue

Signed-off-by: zane-neo <[email protected]>
zane-neo · Jan 29, 2024 · bb6d7b1 · bb6d7b1
1 parent 35f8ef3
commit bb6d7b1
Showing 1 changed file with 3 additions and 0 deletions.
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
@@ -197,6 +197,9 @@ void cleanUpLocalCache(Map<String, Set<String>> 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) {