Skip to content

Commit c12b452

Browse files
committed
* fix wrong lazyloader of rank-bm25
1 parent e15649b commit c12b452

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

data_juicer/tools/op_search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@ def _build_bm25_index(
362362
text = self._get_searchable_text(record, fields)
363363
corpus_tokens.append(self._tokenize(text))
364364

365-
rank_bm25 = LazyLoader("rank-bm25")
365+
rank_bm25 = LazyLoader("rank_bm25", "rank-bm25")
366366
self._bm25_index = rank_bm25.BM25Okapi(corpus_tokens)
367367
self._bm25_records = records
368368
self._bm25_fields_key = tuple(fields)

0 commit comments

Comments
 (0)