From 022869132bbc2ba09d61c232e21fb89bc5c58b55 Mon Sep 17 00:00:00 2001 From: Naveen Tatikonda Date: Tue, 24 Oct 2023 22:55:20 -0500 Subject: [PATCH] Fix Windows CI Signed-off-by: Naveen Tatikonda --- .github/workflows/CI.yml | 1 + jni/include/jni_util.h | 1 + src/test/java/org/opensearch/knn/index/FaissIT.java | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml index d5a6e3f11..84b4f09f9 100644 --- a/.github/workflows/CI.yml +++ b/.github/workflows/CI.yml @@ -108,6 +108,7 @@ jobs: - name: Add MinGW to PATH run: | echo "C:/Users/runneradmin/scoop/apps/mingw/current/bin" >> $env:GITHUB_PATH + Import-Module "$env:ChocolateyInstall\helpers\chocolateyProfile.psm1" refreshenv - name: Download OpenBLAS diff --git a/jni/include/jni_util.h b/jni/include/jni_util.h index 8d6f2b6f6..b4dd44891 100644 --- a/jni/include/jni_util.h +++ b/jni/include/jni_util.h @@ -17,6 +17,7 @@ #include #include #include +#include namespace knn_jni { diff --git a/src/test/java/org/opensearch/knn/index/FaissIT.java b/src/test/java/org/opensearch/knn/index/FaissIT.java index 5db24ea30..9a9f8779f 100644 --- a/src/test/java/org/opensearch/knn/index/FaissIT.java +++ b/src/test/java/org/opensearch/knn/index/FaissIT.java @@ -199,7 +199,7 @@ public void testEndToEnd_whenMethodIsHNSWPQ_thenSucceed() { createBasicKnnIndex(trainingIndexName, trainingFieldName, dimension); ingestDataAndTrainModel(modelId, trainingIndexName, trainingFieldName, dimension, modelDescription, in, trainingDataCount); - assertTrainingSucceeds(modelId, 180, 1000); + assertTrainingSucceeds(modelId, 360, 1000); // Create an index XContentBuilder builder = XContentFactory.jsonBuilder()