Skip to content

Commit faf5b76

Browse files
committed
added tests for faiss_mode
1 parent b8fdc3d commit faf5b76

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_tutorials.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,8 @@ def test_matchingtest(matching_data):
170170
"matching-atc": Matching(metric="atc"),
171171
"matching-att": Matching(metric="att"),
172172
"matching-l2": Matching(distance="l2", metric="att"),
173+
"matching-faiss-auto": Matching(distance="l2", faiss_mode="auto"),
174+
"matching-faiss_base": Matching(distance="mahalanobis", faiss_mode="base"),
173175
}
174176

175177
for test_name in mapping.keys():

0 commit comments

Comments
 (0)