Moving question_answering tests to the new testing scheme. Had to tweak a little some ModelTesterConfig for pipelines. (#13277)

* Moving question_answering tests to the new testing scheme. Had to tweak
a little some ModelTesterConfig for pipelines.

* Removing commented code.
This commit is contained in:
Nicolas Patry
2021-08-26 12:37:55 +02:00
committed by GitHub
parent 4fa1cd995c
commit 55fb88d369
4 changed files with 145 additions and 119 deletions

View File

@@ -147,6 +147,11 @@ class BartModelTester:
pad_token_id=self.pad_token_id,
)
def get_pipeline_config(self):
config = self.get_config()
config.max_position_embeddings = 100
return config
def prepare_config_and_inputs_for_common(self):
config, inputs_dict = self.prepare_config_and_inputs()
return config, inputs_dict