Fixing Build

Signed-off-by: Varun Jain <[email protected]>
opensearch-project · Jan 24, 2024 · 7bccb42 · 7bccb42
1 parent 8cb9f64
commit 7bccb42
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/testFixtures/java/org/opensearch/neuralsearch/BaseNeuralSearchIT.java b/src/testFixtures/java/org/opensearch/neuralsearch/BaseNeuralSearchIT.java
@@ -197,8 +197,6 @@ protected void loadModel(String modelId) throws Exception {
             isComplete = checkComplete(taskQueryResult);
             Thread.sleep(DEFAULT_TASK_RESULT_QUERY_INTERVAL_IN_MILLISECOND);
         }
-
-        pollForModelState(modelId, Set.of(MLModelState.DEPLOYED));
     }
 
     /**
@@ -893,7 +891,7 @@ protected Set<String> findDeployedModels() {
         List<Map<String, Object>> innerHitsMap = (List<Map<String, Object>>) hits.get("hits");
         return innerHitsMap.stream()
             .map(hit -> (Map<String, Object>) hit.get("_source"))
-            .filter(hitsMap -> !Objects.isNull(hitsMap) && hitsMap.containsKey("model_id"))
+            .filter(hitsMap -> Objects.nonNull(hitsMap) && hitsMap.containsKey("model_id"))
             .map(hitsMap -> (String) hitsMap.get("model_id"))
             .collect(Collectors.toSet());
     }