fix test

tests/test_modeling_rag.py

@@ -18,32 +18,20 @@ import copy
 import unittest
 from unittest.mock import patch
 
-import torch
-
-from transformers import (
-    BartConfig,
-    BartForConditionalGeneration,
-    DPRConfig,
-    DPRQuestionEncoder,
-    RagConfig,
-    RagModel,
-    is_torch_available,
-)
-from transformers.file_utils import (
-    cached_property,
-    is_faiss_available,
-    is_nlp_available,
-    is_psutil_available,
-    is_torch_available,
-)
-from transformers.testing_utils import require_torch, slow, torch_device
+from transformers.file_utils import is_faiss_available, is_nlp_available, is_psutil_available, is_torch_available
+from transformers.testing_utils import require_torch, torch_device
 
 from .test_configuration_common import ConfigTester
-from .test_modeling_common import ModelTesterMixin, ids_tensor
+from .test_modeling_common import ids_tensor
 
 
-if is_torch_available() and is_nlp_available() and is_faiss_available() and is_psutil_available():
-    from transformers import AutoModelForMaskedLM, AutoTokenizer, RagConfig, RagSequence, RagToken
+if is_torch_available():
+    import torch
+
+    from transformers import BartConfig, BartForConditionalGeneration, DPRConfig, DPRQuestionEncoder
+
+    if is_nlp_available() and is_faiss_available() and is_psutil_available():
+        from transformers import RagConfig, RagSequence, RagToken
 
 
 class RagModelTester:
@@ -203,7 +191,7 @@ class RagModelTest(unittest.TestCase):
                 with self.assertRaises(
                     AssertionError,
                 ):
-                    model = model_class(
+                    model_class(
                         config=self.model_tester.rag_config,
                         question_encoder=DPRQuestionEncoder(self.model_tester.dpr_config),
                         generator=BartForConditionalGeneration(mismatched_bart_config),