Give option to suppress print statements

PrithivirajDamodaran / FlashRank

Lite & Super-fast re-ranking for your search & retrieval pipelines. Supports SoTA Listwise and Pairwise reranking based on LLMs and cross-encoders and more. Created by Prithivi Da, open for PRs & Collaborations.

Apache License 2.0

595 stars 44 forks source link

Give option to suppress print statements #17

Closed oussama-gourari closed 4 months ago

oussama-gourari commented 4 months ago

Currently there is no option to disable print statements originating from the library, for example "Running pairwise ranking..".

PrithivirajDamodaran commented 4 months ago

Done, play with the log_level keyword to see current or future messages

ranker = Ranker("ms-marco-MultiBERT-L-12", log_level="DEBUG")