From 12f319b4c9bbf9bcd75e2f1f529f767e8e7afcc5 Mon Sep 17 00:00:00 2001 From: Sebastian Husch Lee Date: Wed, 5 Jul 2023 09:37:41 +0200 Subject: [PATCH] Remove deprecated return_table_cell from conftest.py (#5264) --- test/conftest.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) 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, )