Fix null taskId causing model undeploy issue (#1945) (#1947)

Signed-off-by: zane-neo <[email protected]> (cherry picked from commit 7d79bf6) Co-authored-by: zane-neo <[email protected]>
opensearch-project · Jan 29, 2024 · bd8b6ab · bd8b6ab
1 parent 126ed3a
commit bd8b6ab
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) {