Traceback (most recent call last):
File "/lib/python3.8/site-packages/retriv/dense_retriever/dense_retriever.py", line 251, in search
doc_ids = self.map_internal_ids_to_original_ids(doc_ids)
File "/lib/python3.8/site-packages/retriv/base_retriever.py", line 80, in map_internal_ids_to_original_ids
return [self.id_mapping[doc_id] for doc_id in doc_ids]
File "/lib/python3.8/site-packages/retriv/base_retriever.py", line 80, in <listcomp>
return [self.id_mapping[doc_id] for doc_id in doc_ids]
KeyError: -1
Update "/lib/python3.8/site-packages/retriv/base_retriever.py", line 80 to
return [self.id_mapping[doc_id] for doc_id in doc_ids if doc_id != -1]
Update "/lib/python3.8/site-packages/retriv/base_retriever.py", line 80 to
would fix the problem.