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

920 stars 39 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 2 months ago

xhluca commented 2 months 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]]