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.
2021-08-26 12:37:55 +02:00
parent 4fa1cd995c
commit 55fb88d369
4 changed files with 145 additions and 119 deletions
--- a/tests/test_modeling_reformer.py
+++ b/tests/test_modeling_reformer.py
@@ -189,6 +189,7 @@ class ReformerModelTester:
    def get_pipeline_config(self):
        config = self.get_config()
        config.vocab_size = 100
+        config.axial_pos_shape = (4, 25)
        config.is_decoder = False
        return config