diff --git a/tests/test_modeling_albert.py b/tests/test_modeling_albert.py index b1fe4bb6..3e7f17ac 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 6f9f9be6..13eaf53e 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 20b53f6f..b780c05a 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 05585f6d..e6f39c1d 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 96f48791..b059d2a6 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 5e7cc8ad..88b5257d 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 8af1c8d4..15f4c49d 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 74c2d901..ec9940cb 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 f8a56cfc..cd9a2cf2 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 3e58a0b5..cb24039d 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 7f9e98d3..5209719b 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 00ddcc45..af597149 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 d91d4863..6ab7c2b2 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 a04ca7f4..12a42c73 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 5546e7a5..2a67ffbf 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 2a0bfcdb..27e26c0b 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 f8b2ca8e..b2183e26 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 605a1d6d..d5fd21ee 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 7432cc36..7dd89f14 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 cad0aceb..e533087c 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 b432f49e..1d606fd6 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 5048930b..261c592e 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 a66e6a47..a0b0ebad 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 58a89b3e..494c84d5 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 d6adc35b..6a5805c1 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 a4bc104c..e2a30370 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 6036cefd..b932e215 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.