From a45914193ab72da62bb7eaae524eec5a7543c8f1 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 16 Jan 2023 15:09:52 +0100 Subject: [PATCH] Fix `RealmModelIntegrationTest.test_inference_open_qa` (#21136) fix Co-authored-by: ydshieh --- tests/models/realm/test_modeling_realm.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/models/realm/test_modeling_realm.py b/tests/models/realm/test_modeling_realm.py index e084cf5a4e..c087a4dd0d 100644 --- a/tests/models/realm/test_modeling_realm.py +++ b/tests/models/realm/test_modeling_realm.py @@ -480,15 +480,12 @@ class RealmModelIntegrationTest(unittest.TestCase): def test_inference_open_qa(self): from transformers.models.realm.retrieval_realm import RealmRetriever - config = RealmConfig() - tokenizer = RealmTokenizer.from_pretrained("google/realm-orqa-nq-openqa") retriever = RealmRetriever.from_pretrained("google/realm-orqa-nq-openqa") model = RealmForOpenQA.from_pretrained( "google/realm-orqa-nq-openqa", retriever=retriever, - config=config, ) question = "Who is the pioneer in modern computer science?"