yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 21:28:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

bug: didn't clean up model files after running pytest for test_table_text_retriever_training (#3534)

Browse Source 
* Added tmp path to avoid clean up of model files later
This commit is contained in:
Mayank Jobanputra 2022-11-07 10:37:04 +01:00 committed by GitHub
parent 255072d8d5
commit 794fe5ffa4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/nodes/test_retriever.py
View File

@ -424,7 +424,7 @@ def test_table_text_retriever_saving_and_loading(tmp_path, retriever, document_s
@pytest.mark.embedding_dim(128)
def test_table_text_retriever_training(document_store):
def test_table_text_retriever_training(tmp_path, document_store):
retriever = TableTextRetriever(
document_store=document_store,
query_embedding_model="deepset/bert-small-mm_retrieval-question_encoder",
@ -438,11 +438,13 @@ def test_table_text_retriever_training(document_store):
train_filename="sample.json",
n_epochs=1,
n_gpu=0,
save_dir="test_table_text_retriever_train",
save_dir=f"{tmp_path}/test_table_text_retriever_train",
)
# Load trained model
retriever = TableTextRetriever.load(load_dir="test_table_text_retriever_train", document_store=document_store)
retriever = TableTextRetriever.load(
load_dir=f"{tmp_path}/test_table_text_retriever_train", document_store=document_store
)
@pytest.mark.elasticsearch