Update com_spotify_voyager_jni_Index.cpp (#29)

spotify · Oct 17, 2023 · 63b3c81 · 63b3c81
1 parent 22c46d1
commit 63b3c81
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/java/com_spotify_voyager_jni_Index.cpp b/java/com_spotify_voyager_jni_Index.cpp
@@ -728,6 +728,12 @@ void Java_com_spotify_voyager_jni_Index_nativeLoadFromFileWithParameters(
             ") does not match the number of dimensions used in this file (" +
             std::to_string(metadata->getNumDimensions()) + ").");
       }
+
+      setHandle<Index>(
+          env, self,
+          loadTypedIndexFromMetadata(std::move(metadata), inputStream)
+              .release());
+      return;
     }
 
     switch (toStorageDataType(env, storageDataType)) {
@@ -786,6 +792,12 @@ void Java_com_spotify_voyager_jni_Index_nativeLoadFromInputStreamWithParameters(
             ") does not match the number of dimensions used in this file (" +
             std::to_string(metadata->getNumDimensions()) + ").");
       }
+
+      setHandle<Index>(
+          env, self,
+          loadTypedIndexFromMetadata(std::move(metadata), inputStream)
+              .release());
+      return;
     }
 
     switch (toStorageDataType(env, storageDataType)) {
@@ -827,4 +839,4 @@ void Java_com_spotify_voyager_jni_Index_nativeDestructor(JNIEnv *env,
       env->ThrowNew(env->FindClass("java/lang/RuntimeException"), e.what());
     }
   }
-}
+}