yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-01 18:29:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove deprecated return_table_cell from conftest.py (#5264)

Browse Source
This commit is contained in:
Sebastian Husch Lee 2023-07-05 09:37:41 +02:00 committed by GitHub
parent 00efa514ca
commit 12f319b4c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/conftest.py
View File

@ -537,17 +537,16 @@ def reader(request):
@pytest.fixture(params=["tapas_small", "tapas_base", "tapas_scored", "rci"]) @pytest.fixture(params=["tapas_small", "tapas_base", "tapas_scored", "rci"])
def table_reader_and_param(request): def table_reader_and_param(request):
if request.param == "tapas_small": if request.param == "tapas_small":
return TableReader(model_name_or_path="google/tapas-small-finetuned-wtq", return_table_cell=True), request.param return TableReader(model_name_or_path="google/tapas-small-finetuned-wtq"), request.param
elif request.param == "tapas_base": elif request.param == "tapas_base":
return TableReader(model_name_or_path="google/tapas-base-finetuned-wtq", return_table_cell=True), request.param return TableReader(model_name_or_path="google/tapas-base-finetuned-wtq"), request.param
elif request.param == "tapas_scored": elif request.param == "tapas_scored":
return TableReader(model_name_or_path="deepset/tapas-large-nq-hn-reader", return_table_cell=True), request.param return TableReader(model_name_or_path="deepset/tapas-large-nq-hn-reader"), request.param
elif request.param == "rci": elif request.param == "rci":
return ( return (
RCIReader( RCIReader(
row_model_name_or_path="michaelrglass/albert-base-rci-wikisql-row", row_model_name_or_path="michaelrglass/albert-base-rci-wikisql-row",
column_model_name_or_path="michaelrglass/albert-base-rci-wikisql-col", column_model_name_or_path="michaelrglass/albert-base-rci-wikisql-col",
return_table_cell=True,
), ),
request.param, request.param,
) )