Add QSC cleanup to NativeEngines990KnnVectorsFormatTests

Signed-off-by: owenhalpert <[email protected]>
owenhalpert · Jan 21, 2025 · bec302f · bec302f
1 parent 8b14477
commit bec302f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...ava/org/opensearch/knn/index/codec/KNN990Codec/NativeEngines990KnnVectorsFormatTests.java b/...ava/org/opensearch/knn/index/codec/KNN990Codec/NativeEngines990KnnVectorsFormatTests.java
@@ -65,6 +65,7 @@
 import org.opensearch.knn.index.mapper.KNNVectorFieldMapper;
 import org.opensearch.knn.index.engine.KNNEngine;
 import org.opensearch.knn.quantization.enums.ScalarQuantizationType;
+import org.opensearch.knn.quantization.models.quantizationState.QuantizationStateCacheManager;
 
 import java.io.IOException;
 import java.util.Arrays;
@@ -93,6 +94,7 @@ public void tearDown() throws Exception {
         if (dir != null) {
             dir.close();
         }
+        QuantizationStateCacheManager.getInstance().close();
         super.tearDown();
     }