diff --git a/test/conftest.py b/test/conftest.py index 945260418..15ca907de 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -537,17 +537,16 @@ def reader(request): @pytest.fixture(params=["tapas_small", "tapas_base", "tapas_scored", "rci"]) def table_reader_and_param(request): 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": - 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": - 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": return ( RCIReader( row_model_name_or_path="michaelrglass/albert-base-rci-wikisql-row", column_model_name_or_path="michaelrglass/albert-base-rci-wikisql-col", - return_table_cell=True, ), request.param, )