check for NPE for getIsAutoDeploying

Signed-off-by: Xun Zhang <[email protected]>
opensearch-project · May 6, 2024 · 35f06ee · 35f06ee
1 parent 13ff74e
commit 35f06ee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/plugin/src/main/java/org/opensearch/ml/model/MLModelCacheHelper.java b/plugin/src/main/java/org/opensearch/ml/model/MLModelCacheHelper.java
@@ -17,6 +17,7 @@
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.stream.Collectors;
 
+import org.apache.commons.lang3.BooleanUtils;
 import org.opensearch.cluster.service.ClusterService;
 import org.opensearch.common.settings.Settings;
 import org.opensearch.common.util.TokenBucket;
@@ -299,7 +300,7 @@ public synchronized void setIsAutoDeploying(String modelId, Boolean isModelAutoD
      */
     public boolean isAutoDeploying(String modelId) {
         MLModelCache modelCache = modelCaches.get(modelId);
-        return modelCache != null && modelCache.getIsAutoDeploying() == true;
+        return modelCache != null && modelCache.getIsAutoDeploying() != null && BooleanUtils.isTrue(modelCache.getIsAutoDeploying());
     }
 
     /**