SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix examples/rag imports, tests (#7712)

Browse Source
This commit is contained in:
Sam Shleifer
2020-10-14 11:35:00 -04:00
committed by GitHub
parent 890e790e16
commit 8feb0cc967
5 changed files with 42 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/rag/test_distributed_retriever.py
View File

@@ -23,7 +23,7 @@ from transformers.tokenization_roberta import VOCAB_FILES_NAMES as BART_VOCAB_FI
sys.path.append(os.path.join(os.getcwd())) # noqa: E402 # noqa: E402 # isort:skip
from examples.rag.distributed_retriever import RagPyTorchDistributedRetriever # noqa: E402 # isort:skip
from distributed_retriever import RagPyTorchDistributedRetriever # noqa: E402 # isort:skip
def require_distributed_retrieval(test_case):