We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e15649b commit c12b452Copy full SHA for c12b452
1 file changed
data_juicer/tools/op_search.py
@@ -362,7 +362,7 @@ def _build_bm25_index(
362
text = self._get_searchable_text(record, fields)
363
corpus_tokens.append(self._tokenize(text))
364
365
- rank_bm25 = LazyLoader("rank-bm25")
+ rank_bm25 = LazyLoader("rank_bm25", "rank-bm25")
366
self._bm25_index = rank_bm25.BM25Okapi(corpus_tokens)
367
self._bm25_records = records
368
self._bm25_fields_key = tuple(fields)
0 commit comments