castorini / rank_llm

RankLLM is a Python toolkit for reproducible information retrieval research using rerankers, with a focus on listwise reranking.
http://rankllm.ai
Apache License 2.0
312 stars 39 forks source link

minor refactoring and reuqirements.txt update #44

Closed sahel-sh closed 8 months ago

sahel-sh commented 8 months ago

This cl: 1- Changes the retriever.py to always return retrieved results, rather than None in case of pyserini retriver. 2- reranker class does not need the dataset name, I added it as an arg to the write function. 3- Updated requirments.txt to flexible versions, rankzephyr needs more recent transformers version

sahel-sh commented 8 months ago

Updated script files are only format changes