Spaces:
Runtime error
Runtime error
Upload 6 files
Browse files- utils/retriever.py +1 -0
utils/retriever.py
CHANGED
@@ -41,6 +41,7 @@ def get_bm25_search_hits(corpus, sparse_scores, top_n=50):
|
|
41 |
if len(bm25_search) <= top_n:
|
42 |
bm25_search.append(corpus[idx])
|
43 |
indices.append(idx)
|
|
|
44 |
return indices
|
45 |
|
46 |
|
|
|
41 |
if len(bm25_search) <= top_n:
|
42 |
bm25_search.append(corpus[idx])
|
43 |
indices.append(idx)
|
44 |
+
indices = [int(x) for x in indices]
|
45 |
return indices
|
46 |
|
47 |
|