From 6ea88ef5936daa627715bf7c55cea5ae111d1cfa Mon Sep 17 00:00:00 2001
From: a252jain <a252jain@uwaterloo.ca>
Date: Fri, 5 Apr 2024 09:47:11 -0400
Subject: [PATCH] name changes
---
Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift b/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift
index c44b98da..d1e5e35b 100644
--- a/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift
+++ b/Tests/SwiftNLPTests/AllMiniLM_sampleTest.swift
@@ -34,7 +34,7 @@ final class BERT_test: XCTestCase {
var query_embedding: [Float] = []
var embedding_dim: Int = 384
- var model = MiniLMEmbeddings(model_type: "all_miniLM_L6_v2")
+ var model = MiniLMEmbeddings(model_type: "all_MiniLM_L6_v2")
query_embedding = await model.encode(sentence: query[0])!
@@ -46,7 +46,7 @@ final class BERT_test: XCTestCase {
//print(i)
i += 1
} else {
- fatalError("Error occurred1")
+ fatalError("Error occurred!")
}
}
--
GitLab