From 911eb5c6306d4f20782758bc476b4add4ac5d3b9 Mon Sep 17 00:00:00 2001
From: John Mazanec <jmazane@amazon.com>
Date: Wed, 31 Jul 2024 20:34:17 -0400
Subject: [PATCH] Fix package change from k-nn (#855)

Signed-off-by: John Mazanec <jmazane@amazon.com>
---
 .../opensearch/neuralsearch/common/MinClusterVersionUtil.java   | 2 +-
 .../java/org/opensearch/neuralsearch/NeuralSearchTests.java     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java b/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java
index 160b2fa4d..0f5cbefcf 100644
--- a/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java
+++ b/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java
@@ -6,7 +6,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import org.opensearch.Version;
-import org.opensearch.knn.index.IndexUtil;
+import org.opensearch.knn.index.util.IndexUtil;
 import org.opensearch.neuralsearch.util.NeuralSearchClusterUtil;
 
 import java.util.Map;
diff --git a/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java b/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java
index 3220ce8c6..2e9183682 100644
--- a/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java
+++ b/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java
@@ -5,7 +5,7 @@
 package org.opensearch.neuralsearch;
 
 import org.opensearch.knn.common.KNNConstants;
-import org.opensearch.knn.index.util.KNNEngine;
+import org.opensearch.knn.index.engine.KNNEngine;
 import org.opensearch.test.OpenSearchTestCase;
 
 public class NeuralSearchTests extends OpenSearchTestCase {