Add tests for BM25.retrieve in different scenario (tokenized, ids/vocab tuple, object with ids and vocab attributes, ids, strings) - Githubissues

xhluca / bm25s

Fast lexical search implementing BM25 in Python using Numpy, Numba and Scipy

https://bm25s.github.io

MIT License

862 stars 35 forks source link

Add tests for BM25.retrieve in different scenario (tokenized, ids/vocab tuple, object with ids and vocab attributes, ids, strings) #52

Open xhluca opened 1 month ago

xhluca commented 1 month ago

[ ] add tests for BM25.retrieve under each possible condition:
- [ ] Tokenized namedtuple
- [ ] Object with ids and vocab
- [ ] tuple (not Tokenized namedtuple) of ids and vocab
- [ ] ids: List[List[int]]
- [ ] strings: List[List[str]]