examples/rag/test_distributed_retriever.py · cf1c88e0921243e760d306e63a5938e1bac880f3 · chenpangpang / transformers

[RAG] Fix retrieval offset in RAG's HfIndex and better integration tests (#7372) · cf1c88e0

Quentin Lhoest authored Sep 25, 2020



* Fix retrieval offset in RAG's HfIndex

* update slow tests

* style

* fix new test

* style

* add better tests
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

cf1c88e0

test_distributed_retriever.py 6 KB

Replace test_distributed_retriever.py