diff --git a/tests/test_modeling_albert.py b/tests/test_modeling_albert.py index b1fe4bb64e..3e7f17acc5 100644 --- a/tests/test_modeling_albert.py +++ b/tests/test_modeling_albert.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -267,5 +267,5 @@ class AlbertModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = AlbertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = AlbertModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_bart.py b/tests/test_modeling_bart.py index 6f9f9be61d..13eaf53ebe 100644 --- a/tests/test_modeling_bart.py +++ b/tests/test_modeling_bart.py @@ -21,7 +21,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -563,7 +563,7 @@ class BartModelIntegrationTests(unittest.TestCase): def test_model_from_pretrained(self): # Forces 1.6GB download from S3 for each model for model_name in list(BART_PRETRAINED_MODEL_ARCHIVE_MAP.keys()): - model = BartModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = BartModel.from_pretrained(model_name) self.assertIsNotNone(model) @slow diff --git a/tests/test_modeling_bert.py b/tests/test_modeling_bert.py index 20b53f6fad..b780c05ac5 100644 --- a/tests/test_modeling_bert.py +++ b/tests/test_modeling_bert.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -501,5 +501,5 @@ class BertModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(BERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = BertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = BertModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_ctrl.py b/tests/test_modeling_ctrl.py index 05585f6db3..e6f39c1d7c 100644 --- a/tests/test_modeling_ctrl.py +++ b/tests/test_modeling_ctrl.py @@ -19,7 +19,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -211,7 +211,7 @@ class CTRLModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(CTRL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = CTRLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = CTRLModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_distilbert.py b/tests/test_modeling_distilbert.py index 96f4879166..b059d2a6de 100644 --- a/tests/test_modeling_distilbert.py +++ b/tests/test_modeling_distilbert.py @@ -248,5 +248,5 @@ class DistilBertModelTest(ModelTesterMixin, unittest.TestCase): # @slow # def test_model_from_pretrained(self): # for model_name in list(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - # model = DistilBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + # model = DistilBertModel.from_pretrained(model_name) # self.assertIsNotNone(model) diff --git a/tests/test_modeling_electra.py b/tests/test_modeling_electra.py index 5e7cc8ad2f..88b5257d8b 100644 --- a/tests/test_modeling_electra.py +++ b/tests/test_modeling_electra.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -283,5 +283,5 @@ class ElectraModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = ElectraModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = ElectraModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_flaubert.py b/tests/test_modeling_flaubert.py index 8af1c8d40a..15f4c49d5a 100644 --- a/tests/test_modeling_flaubert.py +++ b/tests/test_modeling_flaubert.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -388,5 +388,5 @@ class FlaubertModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(FLAUBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = FlaubertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = FlaubertModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_gpt2.py b/tests/test_modeling_gpt2.py index 74c2d9011f..ec9940cb8f 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -335,7 +335,7 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(GPT2_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = GPT2Model.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = GPT2Model.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_openai.py b/tests/test_modeling_openai.py index f8a56cfcd2..cd9a2cf235 100644 --- a/tests/test_modeling_openai.py +++ b/tests/test_modeling_openai.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -219,7 +219,7 @@ class OpenAIGPTModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = OpenAIGPTModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = OpenAIGPTModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_roberta.py b/tests/test_modeling_roberta.py index 3e58a0b597..cb24039d68 100644 --- a/tests/test_modeling_roberta.py +++ b/tests/test_modeling_roberta.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -274,7 +274,7 @@ class RobertaModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = RobertaModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = RobertaModel.from_pretrained(model_name) self.assertIsNotNone(model) def test_create_position_ids_respects_padding_index(self): diff --git a/tests/test_modeling_t5.py b/tests/test_modeling_t5.py index 7f9e98d383..5209719b59 100644 --- a/tests/test_modeling_t5.py +++ b/tests/test_modeling_t5.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -358,7 +358,7 @@ class T5ModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(T5_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = T5Model.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = T5Model.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_albert.py b/tests/test_modeling_tf_albert.py index 00ddcc45dc..af5971494f 100644 --- a/tests/test_modeling_tf_albert.py +++ b/tests/test_modeling_tf_albert.py @@ -20,7 +20,7 @@ from transformers import AlbertConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -231,5 +231,5 @@ class TFAlbertModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFAlbertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFAlbertModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_bert.py b/tests/test_modeling_tf_bert.py index d91d4863af..6ab7c2b2ca 100644 --- a/tests/test_modeling_tf_bert.py +++ b/tests/test_modeling_tf_bert.py @@ -20,7 +20,7 @@ from transformers import BertConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -313,5 +313,5 @@ class TFBertModelTest(TFModelTesterMixin, unittest.TestCase): def test_model_from_pretrained(self): # for model_name in list(TF_BERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: for model_name in ["bert-base-uncased"]: - model = TFBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFBertModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_ctrl.py b/tests/test_modeling_tf_ctrl.py index a04ca7f466..12a42c7314 100644 --- a/tests/test_modeling_tf_ctrl.py +++ b/tests/test_modeling_tf_ctrl.py @@ -20,7 +20,7 @@ from transformers import CTRLConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -201,7 +201,7 @@ class TFCTRLModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_CTRL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFCTRLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFCTRLModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_distilbert.py b/tests/test_modeling_tf_distilbert.py index 5546e7a5b8..2a67ffbfc9 100644 --- a/tests/test_modeling_tf_distilbert.py +++ b/tests/test_modeling_tf_distilbert.py @@ -219,5 +219,5 @@ class TFDistilBertModelTest(TFModelTesterMixin, unittest.TestCase): # @slow # def test_model_from_pretrained(self): # for model_name in list(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - # model = DistilBertModesss.from_pretrained(model_name, cache_dir=CACHE_DIR) + # model = DistilBertModesss.from_pretrained(model_name) # self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_electra.py b/tests/test_modeling_tf_electra.py index 2a0bfcdbb5..27e26c0b26 100644 --- a/tests/test_modeling_tf_electra.py +++ b/tests/test_modeling_tf_electra.py @@ -20,7 +20,7 @@ from transformers import ElectraConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -223,5 +223,5 @@ class TFElectraModelTest(TFModelTesterMixin, unittest.TestCase): def test_model_from_pretrained(self): # for model_name in list(TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: for model_name in ["electra-small-discriminator"]: - model = TFElectraModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFElectraModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_gpt2.py b/tests/test_modeling_tf_gpt2.py index f8b2ca8e3b..b2183e26ca 100644 --- a/tests/test_modeling_tf_gpt2.py +++ b/tests/test_modeling_tf_gpt2.py @@ -20,7 +20,7 @@ from transformers import GPT2Config, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -324,7 +324,7 @@ class TFGPT2ModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_GPT2_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFGPT2Model.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFGPT2Model.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_openai_gpt.py b/tests/test_modeling_tf_openai_gpt.py index 605a1d6d45..d5fd21ee7e 100644 --- a/tests/test_modeling_tf_openai_gpt.py +++ b/tests/test_modeling_tf_openai_gpt.py @@ -20,7 +20,7 @@ from transformers import OpenAIGPTConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -236,7 +236,7 @@ class TFOpenAIGPTModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFOpenAIGPTModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFOpenAIGPTModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_roberta.py b/tests/test_modeling_tf_roberta.py index 7432cc3607..7dd89f14d4 100644 --- a/tests/test_modeling_tf_roberta.py +++ b/tests/test_modeling_tf_roberta.py @@ -20,7 +20,7 @@ from transformers import RobertaConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -233,7 +233,7 @@ class TFRobertaModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFRobertaModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFRobertaModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_t5.py b/tests/test_modeling_tf_t5.py index cad0aceb1c..e533087c1c 100644 --- a/tests/test_modeling_tf_t5.py +++ b/tests/test_modeling_tf_t5.py @@ -20,7 +20,7 @@ from transformers import T5Config, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -260,7 +260,7 @@ class TFT5ModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in ["t5-small"]: - model = TFT5Model.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFT5Model.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_transfo_xl.py b/tests/test_modeling_tf_transfo_xl.py index b432f49e3c..1d606fd61c 100644 --- a/tests/test_modeling_tf_transfo_xl.py +++ b/tests/test_modeling_tf_transfo_xl.py @@ -21,7 +21,7 @@ from transformers import TransfoXLConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -210,7 +210,7 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFTransfoXLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFTransfoXLModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_xlm.py b/tests/test_modeling_tf_xlm.py index 5048930b56..261c592eda 100644 --- a/tests/test_modeling_tf_xlm.py +++ b/tests/test_modeling_tf_xlm.py @@ -20,7 +20,7 @@ from transformers import is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -309,7 +309,7 @@ class TFXLMModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_XLM_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFXLMModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFXLMModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_tf_xlnet.py b/tests/test_modeling_tf_xlnet.py index a66e6a47c1..a0b0ebada7 100644 --- a/tests/test_modeling_tf_xlnet.py +++ b/tests/test_modeling_tf_xlnet.py @@ -21,7 +21,7 @@ from transformers import XLNetConfig, is_tf_available from .test_configuration_common import ConfigTester from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_tf, slow +from .utils import require_tf, slow if is_tf_available(): @@ -411,7 +411,7 @@ class TFXLNetModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TF_XLNET_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TFXLNetModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TFXLNetModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_transfo_xl.py b/tests/test_modeling_transfo_xl.py index 58a89b3ec7..494c84d513 100644 --- a/tests/test_modeling_transfo_xl.py +++ b/tests/test_modeling_transfo_xl.py @@ -21,7 +21,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -210,7 +210,7 @@ class TransfoXLModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = TransfoXLModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = TransfoXLModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_xlm.py b/tests/test_modeling_xlm.py index d6adc35bf6..6a5805c1ae 100644 --- a/tests/test_modeling_xlm.py +++ b/tests/test_modeling_xlm.py @@ -20,7 +20,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -426,7 +426,7 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(XLM_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = XLMModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = XLMModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/test_modeling_xlnet.py b/tests/test_modeling_xlnet.py index a4bc104c18..e2a3037053 100644 --- a/tests/test_modeling_xlnet.py +++ b/tests/test_modeling_xlnet.py @@ -21,7 +21,7 @@ from transformers import is_torch_available from .test_configuration_common import ConfigTester from .test_modeling_common import ModelTesterMixin, ids_tensor -from .utils import CACHE_DIR, require_torch, slow, torch_device +from .utils import require_torch, slow, torch_device if is_torch_available(): @@ -509,7 +509,7 @@ class XLNetModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in list(XLNET_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]: - model = XLNetModel.from_pretrained(model_name, cache_dir=CACHE_DIR) + model = XLNetModel.from_pretrained(model_name) self.assertIsNotNone(model) diff --git a/tests/utils.py b/tests/utils.py index 6036cefd9d..b932e2154a 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,13 +1,10 @@ import os -import tempfile import unittest from distutils.util import strtobool from transformers.file_utils import _tf_available, _torch_available -CACHE_DIR = os.path.join(tempfile.gettempdir(), "transformers_test") - SMALL_MODEL_IDENTIFIER = "julien-c/bert-xsmall-dummy" DUMMY_UNKWOWN_IDENTIFIER = "julien-c/dummy-unknown" # Used to test Auto{Config, Model, Tokenizer} model_type detection.