diff --git a/extensions/fasttext/src/main/native/ai_djl_fasttext_jni_FastTextLibrary.cc b/extensions/fasttext/src/main/native/ai_djl_fasttext_jni_FastTextLibrary.cc index 8a14c1078b27..faad19a86a50 100644 --- a/extensions/fasttext/src/main/native/ai_djl_fasttext_jni_FastTextLibrary.cc +++ b/extensions/fasttext/src/main/native/ai_djl_fasttext_jni_FastTextLibrary.cc @@ -116,14 +116,15 @@ JNIEXPORT jint JNICALL Java_ai_djl_fasttext_jni_FastTextLibrary_predictProba( fasttext_ptr->predictLine(in, predictions, top_k, 0.0); int size = predictions.size(); - jclass java_lang_Float = static_cast(env->NewGlobalRef(env->FindClass("java/lang/Float"))); - jmethodID java_lang_Float_ = env->GetMethodID(java_lang_Float, "", "(F)V"); + jclass java_lang_Float = static_cast(env->NewGlobalRef(env->FindClass("java/lang/Float"))); + jmethodID java_lang_Float_ = env->GetMethodID(java_lang_Float, "", "(F)V"); jclass java_util_ArrayList = static_cast(env->NewGlobalRef(env->FindClass("java/util/ArrayList"))); jmethodID java_util_ArrayList_add = env->GetMethodID(java_util_ArrayList, "add", "(Ljava/lang/Object;)Z"); for (int i = 0; i < size; ++i) { std::pair pair = predictions[i]; - env->CallBooleanMethod(jclasses, java_util_ArrayList_add, env->NewStringUTF(pair.second.c_str())); - env->CallBooleanMethod(jprob, java_util_ArrayList_add, env->NewObject(java_lang_Float, java_lang_Float_, pair.first)); + env->CallBooleanMethod(jclasses, java_util_ArrayList_add, env->NewStringUTF(pair.second.c_str())); + env->CallBooleanMethod( + jprob, java_util_ArrayList_add, env->NewObject(java_lang_Float, java_lang_Float_, pair.first)); } return size; diff --git a/extensions/sentencepiece/src/main/native/ai_djl_sentencepiece_jni_SentencePieceLibrary.cc b/extensions/sentencepiece/src/main/native/ai_djl_sentencepiece_jni_SentencePieceLibrary.cc index de0d942616d9..dd2c80ed599f 100644 --- a/extensions/sentencepiece/src/main/native/ai_djl_sentencepiece_jni_SentencePieceLibrary.cc +++ b/extensions/sentencepiece/src/main/native/ai_djl_sentencepiece_jni_SentencePieceLibrary.cc @@ -38,13 +38,13 @@ JNIEXPORT void JNICALL Java_ai_djl_sentencepiece_jni_SentencePieceLibrary_loadMo } JNIEXPORT void JNICALL Java_ai_djl_sentencepiece_jni_SentencePieceLibrary_loadModelFromBytes( - JNIEnv* env, jobject jthis, jlong jhandle, jbyteArray jserialized) { - auto* processor_ptr = reinterpret_cast(jhandle); - int length = env->GetArrayLength(jserialized); - std::vector buff(length, 0); - env->GetByteArrayRegion(jserialized, 0, length, reinterpret_cast(buff.data())); - std::string serialized(buff.data(), buff.size()); - CheckStatus(env, processor_ptr->LoadFromSerializedProto(serialized)); + JNIEnv* env, jobject jthis, jlong jhandle, jbyteArray jserialized) { + auto* processor_ptr = reinterpret_cast(jhandle); + int length = env->GetArrayLength(jserialized); + std::vector buff(length, 0); + env->GetByteArrayRegion(jserialized, 0, length, reinterpret_cast(buff.data())); + std::string serialized(buff.data(), buff.size()); + CheckStatus(env, processor_ptr->LoadFromSerializedProto(serialized)); } JNIEXPORT void JNICALL Java_ai_djl_sentencepiece_jni_SentencePieceLibrary_deleteSentencePieceProcessor(