From c962c2adbff678ae6d2e98378bed5b8d1a9831d9 Mon Sep 17 00:00:00 2001 From: Jonas Kuball Date: Fri, 21 Jan 2022 13:30:12 +0100 Subject: [PATCH] Adds missing module_specs for usages of _LazyModule (#15230) * Add missing __spec__ for transformers.models.auto * Moves the __spec__-test to the UnitTest class * Adds module_spec to all instances of _LazyModule * Refactors an old test from pytest to unittest --- src/transformers/models/albert/__init__.py | 2 +- src/transformers/models/auto/__init__.py | 2 +- src/transformers/models/bart/__init__.py | 2 +- src/transformers/models/barthez/__init__.py | 2 +- src/transformers/models/bartpho/__init__.py | 2 +- src/transformers/models/beit/__init__.py | 2 +- src/transformers/models/bert/__init__.py | 2 +- src/transformers/models/bert_generation/__init__.py | 2 +- src/transformers/models/bert_japanese/__init__.py | 2 +- src/transformers/models/bertweet/__init__.py | 2 +- src/transformers/models/big_bird/__init__.py | 2 +- src/transformers/models/bigbird_pegasus/__init__.py | 2 +- src/transformers/models/blenderbot/__init__.py | 2 +- src/transformers/models/blenderbot_small/__init__.py | 2 +- src/transformers/models/byt5/__init__.py | 2 +- src/transformers/models/camembert/__init__.py | 2 +- src/transformers/models/canine/__init__.py | 2 +- src/transformers/models/clip/__init__.py | 2 +- src/transformers/models/convbert/__init__.py | 2 +- src/transformers/models/cpm/__init__.py | 2 +- src/transformers/models/ctrl/__init__.py | 2 +- src/transformers/models/deberta/__init__.py | 2 +- src/transformers/models/deberta_v2/__init__.py | 2 +- src/transformers/models/deit/__init__.py | 2 +- src/transformers/models/detr/__init__.py | 2 +- src/transformers/models/distilbert/__init__.py | 2 +- src/transformers/models/dpr/__init__.py | 2 +- src/transformers/models/electra/__init__.py | 2 +- src/transformers/models/encoder_decoder/__init__.py | 2 +- src/transformers/models/flaubert/__init__.py | 2 +- src/transformers/models/fnet/__init__.py | 2 +- src/transformers/models/fsmt/__init__.py | 2 +- src/transformers/models/funnel/__init__.py | 2 +- src/transformers/models/gpt2/__init__.py | 2 +- src/transformers/models/gpt_neo/__init__.py | 2 +- src/transformers/models/gptj/__init__.py | 2 +- src/transformers/models/herbert/__init__.py | 2 +- src/transformers/models/hubert/__init__.py | 2 +- src/transformers/models/ibert/__init__.py | 2 +- src/transformers/models/imagegpt/__init__.py | 2 +- src/transformers/models/layoutlm/__init__.py | 2 +- src/transformers/models/layoutlmv2/__init__.py | 2 +- src/transformers/models/layoutxlm/__init__.py | 2 +- src/transformers/models/led/__init__.py | 2 +- src/transformers/models/longformer/__init__.py | 2 +- src/transformers/models/luke/__init__.py | 2 +- src/transformers/models/lxmert/__init__.py | 2 +- src/transformers/models/m2m_100/__init__.py | 2 +- src/transformers/models/marian/__init__.py | 2 +- src/transformers/models/mbart/__init__.py | 2 +- src/transformers/models/mbart50/__init__.py | 2 +- src/transformers/models/megatron_bert/__init__.py | 2 +- src/transformers/models/mluke/__init__.py | 2 +- src/transformers/models/mmbt/__init__.py | 2 +- src/transformers/models/mobilebert/__init__.py | 2 +- src/transformers/models/mpnet/__init__.py | 2 +- src/transformers/models/mt5/__init__.py | 1 + src/transformers/models/nystromformer/__init__.py | 2 +- src/transformers/models/openai/__init__.py | 2 +- src/transformers/models/pegasus/__init__.py | 2 +- src/transformers/models/perceiver/__init__.py | 2 +- src/transformers/models/phobert/__init__.py | 2 +- src/transformers/models/prophetnet/__init__.py | 2 +- src/transformers/models/qdqbert/__init__.py | 2 +- src/transformers/models/rag/__init__.py | 2 +- src/transformers/models/realm/__init__.py | 2 +- src/transformers/models/reformer/__init__.py | 2 +- src/transformers/models/rembert/__init__.py | 2 +- src/transformers/models/retribert/__init__.py | 2 +- src/transformers/models/roberta/__init__.py | 2 +- src/transformers/models/roformer/__init__.py | 2 +- src/transformers/models/segformer/__init__.py | 2 +- src/transformers/models/sew/__init__.py | 2 +- src/transformers/models/sew_d/__init__.py | 2 +- .../models/speech_encoder_decoder/__init__.py | 2 +- src/transformers/models/speech_to_text/__init__.py | 2 +- src/transformers/models/speech_to_text_2/__init__.py | 2 +- src/transformers/models/splinter/__init__.py | 2 +- src/transformers/models/squeezebert/__init__.py | 2 +- src/transformers/models/t5/__init__.py | 2 +- src/transformers/models/tapas/__init__.py | 2 +- src/transformers/models/transfo_xl/__init__.py | 2 +- src/transformers/models/trocr/__init__.py | 2 +- src/transformers/models/unispeech/__init__.py | 2 +- src/transformers/models/unispeech_sat/__init__.py | 2 +- .../models/vision_encoder_decoder/__init__.py | 2 +- src/transformers/models/visual_bert/__init__.py | 2 +- src/transformers/models/vit/__init__.py | 2 +- src/transformers/models/vit_mae/__init__.py | 2 +- src/transformers/models/wav2vec2/__init__.py | 2 +- src/transformers/models/wav2vec2_phoneme/__init__.py | 2 +- src/transformers/models/wav2vec2_with_lm/__init__.py | 2 +- src/transformers/models/wavlm/__init__.py | 2 +- src/transformers/models/xlm/__init__.py | 2 +- src/transformers/models/xlm_prophetnet/__init__.py | 2 +- src/transformers/models/xlm_roberta/__init__.py | 2 +- src/transformers/models/xlnet/__init__.py | 2 +- src/transformers/onnx/__init__.py | 2 +- .../__init__.py | 2 +- tests/test_configuration_auto.py | 6 ++++++ tests/test_file_utils.py | 8 +++++--- 101 files changed, 110 insertions(+), 101 deletions(-) diff --git a/src/transformers/models/albert/__init__.py b/src/transformers/models/albert/__init__.py index 183140c6e0..6df6ac17f1 100644 --- a/src/transformers/models/albert/__init__.py +++ b/src/transformers/models/albert/__init__.py @@ -129,4 +129,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/auto/__init__.py b/src/transformers/models/auto/__init__.py index 16848efea8..89d1c73b77 100644 --- a/src/transformers/models/auto/__init__.py +++ b/src/transformers/models/auto/__init__.py @@ -248,4 +248,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bart/__init__.py b/src/transformers/models/bart/__init__.py index a8ddcecc41..ffcf517416 100644 --- a/src/transformers/models/bart/__init__.py +++ b/src/transformers/models/bart/__init__.py @@ -86,4 +86,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/barthez/__init__.py b/src/transformers/models/barthez/__init__.py index 039af6a9e5..f7ec14c5dd 100644 --- a/src/transformers/models/barthez/__init__.py +++ b/src/transformers/models/barthez/__init__.py @@ -41,4 +41,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bartpho/__init__.py b/src/transformers/models/bartpho/__init__.py index 9d57a21845..e12b418036 100644 --- a/src/transformers/models/bartpho/__init__.py +++ b/src/transformers/models/bartpho/__init__.py @@ -33,4 +33,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/beit/__init__.py b/src/transformers/models/beit/__init__.py index c9e311d7cf..52d5b7e2b9 100644 --- a/src/transformers/models/beit/__init__.py +++ b/src/transformers/models/beit/__init__.py @@ -75,4 +75,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bert/__init__.py b/src/transformers/models/bert/__init__.py index 9bcf372282..8f3ea0336f 100644 --- a/src/transformers/models/bert/__init__.py +++ b/src/transformers/models/bert/__init__.py @@ -133,4 +133,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bert_generation/__init__.py b/src/transformers/models/bert_generation/__init__.py index cdfc1fed59..32219a9649 100644 --- a/src/transformers/models/bert_generation/__init__.py +++ b/src/transformers/models/bert_generation/__init__.py @@ -54,4 +54,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bert_japanese/__init__.py b/src/transformers/models/bert_japanese/__init__.py index 7bd18cd0a6..0026a5803d 100644 --- a/src/transformers/models/bert_japanese/__init__.py +++ b/src/transformers/models/bert_japanese/__init__.py @@ -32,4 +32,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bertweet/__init__.py b/src/transformers/models/bertweet/__init__.py index aeee5a2410..675031d9fb 100644 --- a/src/transformers/models/bertweet/__init__.py +++ b/src/transformers/models/bertweet/__init__.py @@ -32,4 +32,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/big_bird/__init__.py b/src/transformers/models/big_bird/__init__.py index 4804f838af..cdff90c8da 100644 --- a/src/transformers/models/big_bird/__init__.py +++ b/src/transformers/models/big_bird/__init__.py @@ -105,4 +105,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/bigbird_pegasus/__init__.py b/src/transformers/models/bigbird_pegasus/__init__.py index dfaef8fe4d..1cee0664ff 100644 --- a/src/transformers/models/bigbird_pegasus/__init__.py +++ b/src/transformers/models/bigbird_pegasus/__init__.py @@ -54,4 +54,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/blenderbot/__init__.py b/src/transformers/models/blenderbot/__init__.py index 59a6b1b901..ec04b1c076 100644 --- a/src/transformers/models/blenderbot/__init__.py +++ b/src/transformers/models/blenderbot/__init__.py @@ -88,4 +88,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/blenderbot_small/__init__.py b/src/transformers/models/blenderbot_small/__init__.py index 6442c08bb9..0f5a2a5bfb 100644 --- a/src/transformers/models/blenderbot_small/__init__.py +++ b/src/transformers/models/blenderbot_small/__init__.py @@ -84,4 +84,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/byt5/__init__.py b/src/transformers/models/byt5/__init__.py index 1d00a73c16..83ea9bba61 100644 --- a/src/transformers/models/byt5/__init__.py +++ b/src/transformers/models/byt5/__init__.py @@ -31,4 +31,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/camembert/__init__.py b/src/transformers/models/camembert/__init__.py index 6079655c38..3eb99ad948 100644 --- a/src/transformers/models/camembert/__init__.py +++ b/src/transformers/models/camembert/__init__.py @@ -96,4 +96,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/canine/__init__.py b/src/transformers/models/canine/__init__.py index 11ee85ddc7..1773dbf0ac 100644 --- a/src/transformers/models/canine/__init__.py +++ b/src/transformers/models/canine/__init__.py @@ -60,4 +60,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/clip/__init__.py b/src/transformers/models/clip/__init__.py index 310f22cfc6..9c0e8d6267 100644 --- a/src/transformers/models/clip/__init__.py +++ b/src/transformers/models/clip/__init__.py @@ -111,4 +111,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/convbert/__init__.py b/src/transformers/models/convbert/__init__.py index 35ffc44f19..4195dda68a 100644 --- a/src/transformers/models/convbert/__init__.py +++ b/src/transformers/models/convbert/__init__.py @@ -95,4 +95,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/cpm/__init__.py b/src/transformers/models/cpm/__init__.py index 9f80c2dba4..e160402e24 100644 --- a/src/transformers/models/cpm/__init__.py +++ b/src/transformers/models/cpm/__init__.py @@ -40,4 +40,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/ctrl/__init__.py b/src/transformers/models/ctrl/__init__.py index 7a15fe7174..fafc6f0c7d 100644 --- a/src/transformers/models/ctrl/__init__.py +++ b/src/transformers/models/ctrl/__init__.py @@ -70,4 +70,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/deberta/__init__.py b/src/transformers/models/deberta/__init__.py index 731fde4eae..1126704b9e 100644 --- a/src/transformers/models/deberta/__init__.py +++ b/src/transformers/models/deberta/__init__.py @@ -85,4 +85,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/deberta_v2/__init__.py b/src/transformers/models/deberta_v2/__init__.py index 90d4267953..1b2cb9e335 100644 --- a/src/transformers/models/deberta_v2/__init__.py +++ b/src/transformers/models/deberta_v2/__init__.py @@ -78,4 +78,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/deit/__init__.py b/src/transformers/models/deit/__init__.py index f0ac888f50..a637cde38b 100644 --- a/src/transformers/models/deit/__init__.py +++ b/src/transformers/models/deit/__init__.py @@ -56,4 +56,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/detr/__init__.py b/src/transformers/models/detr/__init__.py index d2ef0c24f3..bbde97ccbc 100644 --- a/src/transformers/models/detr/__init__.py +++ b/src/transformers/models/detr/__init__.py @@ -56,4 +56,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/distilbert/__init__.py b/src/transformers/models/distilbert/__init__.py index 3be3dda0fa..a6215432af 100644 --- a/src/transformers/models/distilbert/__init__.py +++ b/src/transformers/models/distilbert/__init__.py @@ -120,4 +120,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/dpr/__init__.py b/src/transformers/models/dpr/__init__.py index 24358c1134..4c3074543f 100644 --- a/src/transformers/models/dpr/__init__.py +++ b/src/transformers/models/dpr/__init__.py @@ -113,4 +113,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/electra/__init__.py b/src/transformers/models/electra/__init__.py index 794abb2154..1aad02a412 100644 --- a/src/transformers/models/electra/__init__.py +++ b/src/transformers/models/electra/__init__.py @@ -120,4 +120,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/encoder_decoder/__init__.py b/src/transformers/models/encoder_decoder/__init__.py index a5eafdf251..400ef62282 100644 --- a/src/transformers/models/encoder_decoder/__init__.py +++ b/src/transformers/models/encoder_decoder/__init__.py @@ -49,4 +49,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/flaubert/__init__.py b/src/transformers/models/flaubert/__init__.py index 6d9f87d73f..538a9d6a8c 100644 --- a/src/transformers/models/flaubert/__init__.py +++ b/src/transformers/models/flaubert/__init__.py @@ -82,4 +82,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/fnet/__init__.py b/src/transformers/models/fnet/__init__.py index a28047df9e..8a163e19be 100644 --- a/src/transformers/models/fnet/__init__.py +++ b/src/transformers/models/fnet/__init__.py @@ -70,4 +70,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/fsmt/__init__.py b/src/transformers/models/fsmt/__init__.py index 933addca94..9e633dbc7f 100644 --- a/src/transformers/models/fsmt/__init__.py +++ b/src/transformers/models/fsmt/__init__.py @@ -40,4 +40,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/funnel/__init__.py b/src/transformers/models/funnel/__init__.py index fdb78d944b..e520acae4f 100644 --- a/src/transformers/models/funnel/__init__.py +++ b/src/transformers/models/funnel/__init__.py @@ -99,4 +99,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/gpt2/__init__.py b/src/transformers/models/gpt2/__init__.py index 7169ddc63f..d439d4bc39 100644 --- a/src/transformers/models/gpt2/__init__.py +++ b/src/transformers/models/gpt2/__init__.py @@ -91,4 +91,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/gpt_neo/__init__.py b/src/transformers/models/gpt_neo/__init__.py index 3112d9a36e..70793af636 100644 --- a/src/transformers/models/gpt_neo/__init__.py +++ b/src/transformers/models/gpt_neo/__init__.py @@ -62,4 +62,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/gptj/__init__.py b/src/transformers/models/gptj/__init__.py index 29f1a2baa8..076426dc4b 100644 --- a/src/transformers/models/gptj/__init__.py +++ b/src/transformers/models/gptj/__init__.py @@ -61,4 +61,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/herbert/__init__.py b/src/transformers/models/herbert/__init__.py index 297a82f9aa..d5426c8866 100644 --- a/src/transformers/models/herbert/__init__.py +++ b/src/transformers/models/herbert/__init__.py @@ -38,4 +38,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/hubert/__init__.py b/src/transformers/models/hubert/__init__.py index 1d88b206a4..5b24698c38 100644 --- a/src/transformers/models/hubert/__init__.py +++ b/src/transformers/models/hubert/__init__.py @@ -68,4 +68,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/ibert/__init__.py b/src/transformers/models/ibert/__init__.py index 9ef9780807..0701a44c06 100644 --- a/src/transformers/models/ibert/__init__.py +++ b/src/transformers/models/ibert/__init__.py @@ -55,4 +55,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/imagegpt/__init__.py b/src/transformers/models/imagegpt/__init__.py index 4fc9496ee9..2fbc923b21 100644 --- a/src/transformers/models/imagegpt/__init__.py +++ b/src/transformers/models/imagegpt/__init__.py @@ -58,4 +58,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/layoutlm/__init__.py b/src/transformers/models/layoutlm/__init__.py index 3f34012be7..b20cfdfa8f 100644 --- a/src/transformers/models/layoutlm/__init__.py +++ b/src/transformers/models/layoutlm/__init__.py @@ -83,4 +83,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/layoutlmv2/__init__.py b/src/transformers/models/layoutlmv2/__init__.py index c75d075e59..c5b628cb00 100644 --- a/src/transformers/models/layoutlmv2/__init__.py +++ b/src/transformers/models/layoutlmv2/__init__.py @@ -68,4 +68,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/layoutxlm/__init__.py b/src/transformers/models/layoutxlm/__init__.py index 3bfb2bf58b..2740f9fd8a 100644 --- a/src/transformers/models/layoutxlm/__init__.py +++ b/src/transformers/models/layoutxlm/__init__.py @@ -51,4 +51,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/led/__init__.py b/src/transformers/models/led/__init__.py index 56f1c92385..9a7cb5e863 100644 --- a/src/transformers/models/led/__init__.py +++ b/src/transformers/models/led/__init__.py @@ -66,4 +66,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/longformer/__init__.py b/src/transformers/models/longformer/__init__.py index 959887f479..dc65fe286a 100644 --- a/src/transformers/models/longformer/__init__.py +++ b/src/transformers/models/longformer/__init__.py @@ -100,4 +100,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/luke/__init__.py b/src/transformers/models/luke/__init__.py index 8c6275d12a..29b963a834 100644 --- a/src/transformers/models/luke/__init__.py +++ b/src/transformers/models/luke/__init__.py @@ -56,4 +56,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/lxmert/__init__.py b/src/transformers/models/lxmert/__init__.py index ddbf01c245..1d98d2a10c 100644 --- a/src/transformers/models/lxmert/__init__.py +++ b/src/transformers/models/lxmert/__init__.py @@ -82,4 +82,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/m2m_100/__init__.py b/src/transformers/models/m2m_100/__init__.py index 111e3fb336..6e79e9791b 100644 --- a/src/transformers/models/m2m_100/__init__.py +++ b/src/transformers/models/m2m_100/__init__.py @@ -51,4 +51,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/marian/__init__.py b/src/transformers/models/marian/__init__.py index 3348965a80..1094cb8b9d 100644 --- a/src/transformers/models/marian/__init__.py +++ b/src/transformers/models/marian/__init__.py @@ -72,4 +72,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mbart/__init__.py b/src/transformers/models/mbart/__init__.py index 613c90afbe..cad664c140 100644 --- a/src/transformers/models/mbart/__init__.py +++ b/src/transformers/models/mbart/__init__.py @@ -100,4 +100,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mbart50/__init__.py b/src/transformers/models/mbart50/__init__.py index 299821063d..bb8b11d1f6 100644 --- a/src/transformers/models/mbart50/__init__.py +++ b/src/transformers/models/mbart50/__init__.py @@ -39,4 +39,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/megatron_bert/__init__.py b/src/transformers/models/megatron_bert/__init__.py index 78cb30f3ac..2835d7a5b1 100644 --- a/src/transformers/models/megatron_bert/__init__.py +++ b/src/transformers/models/megatron_bert/__init__.py @@ -60,4 +60,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mluke/__init__.py b/src/transformers/models/mluke/__init__.py index 8982d219f6..9842794def 100644 --- a/src/transformers/models/mluke/__init__.py +++ b/src/transformers/models/mluke/__init__.py @@ -35,4 +35,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mmbt/__init__.py b/src/transformers/models/mmbt/__init__.py index 02f5864fc1..c1daba13a2 100644 --- a/src/transformers/models/mmbt/__init__.py +++ b/src/transformers/models/mmbt/__init__.py @@ -38,4 +38,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mobilebert/__init__.py b/src/transformers/models/mobilebert/__init__.py index bd8f124ecc..e460e348f7 100644 --- a/src/transformers/models/mobilebert/__init__.py +++ b/src/transformers/models/mobilebert/__init__.py @@ -102,4 +102,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mpnet/__init__.py b/src/transformers/models/mpnet/__init__.py index 8bbc4ba273..0ba9e8d6eb 100644 --- a/src/transformers/models/mpnet/__init__.py +++ b/src/transformers/models/mpnet/__init__.py @@ -94,4 +94,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/mt5/__init__.py b/src/transformers/models/mt5/__init__.py index 481196e7ba..99ab1f4b6a 100644 --- a/src/transformers/models/mt5/__init__.py +++ b/src/transformers/models/mt5/__init__.py @@ -76,4 +76,5 @@ else: globals()["__file__"], _import_structure, extra_objects={"MT5Tokenizer": MT5Tokenizer, "MT5TokenizerFast": MT5TokenizerFast}, + module_spec=__spec__, ) diff --git a/src/transformers/models/nystromformer/__init__.py b/src/transformers/models/nystromformer/__init__.py index 9241df25a2..d4d3872be7 100644 --- a/src/transformers/models/nystromformer/__init__.py +++ b/src/transformers/models/nystromformer/__init__.py @@ -59,4 +59,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/openai/__init__.py b/src/transformers/models/openai/__init__.py index 478fc38412..b077620361 100644 --- a/src/transformers/models/openai/__init__.py +++ b/src/transformers/models/openai/__init__.py @@ -84,4 +84,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/pegasus/__init__.py b/src/transformers/models/pegasus/__init__.py index c6b690e7a6..bc7489c883 100644 --- a/src/transformers/models/pegasus/__init__.py +++ b/src/transformers/models/pegasus/__init__.py @@ -92,4 +92,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/perceiver/__init__.py b/src/transformers/models/perceiver/__init__.py index f066c6cacf..f226d8df0c 100644 --- a/src/transformers/models/perceiver/__init__.py +++ b/src/transformers/models/perceiver/__init__.py @@ -69,4 +69,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/phobert/__init__.py b/src/transformers/models/phobert/__init__.py index 824a6e3c95..13c750964a 100644 --- a/src/transformers/models/phobert/__init__.py +++ b/src/transformers/models/phobert/__init__.py @@ -32,4 +32,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/prophetnet/__init__.py b/src/transformers/models/prophetnet/__init__.py index 5774845cfb..45b7d9a5d4 100644 --- a/src/transformers/models/prophetnet/__init__.py +++ b/src/transformers/models/prophetnet/__init__.py @@ -56,4 +56,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/qdqbert/__init__.py b/src/transformers/models/qdqbert/__init__.py index fc03327142..c403795e54 100644 --- a/src/transformers/models/qdqbert/__init__.py +++ b/src/transformers/models/qdqbert/__init__.py @@ -64,4 +64,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/rag/__init__.py b/src/transformers/models/rag/__init__.py index 2d86e44ced..6154997879 100644 --- a/src/transformers/models/rag/__init__.py +++ b/src/transformers/models/rag/__init__.py @@ -63,4 +63,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/realm/__init__.py b/src/transformers/models/realm/__init__.py index 41fcce7be7..b44ddeedee 100644 --- a/src/transformers/models/realm/__init__.py +++ b/src/transformers/models/realm/__init__.py @@ -66,4 +66,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/reformer/__init__.py b/src/transformers/models/reformer/__init__.py index 888d4748b6..246641bdd4 100644 --- a/src/transformers/models/reformer/__init__.py +++ b/src/transformers/models/reformer/__init__.py @@ -70,4 +70,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/rembert/__init__.py b/src/transformers/models/rembert/__init__.py index a9b4083b97..03f3907b5d 100644 --- a/src/transformers/models/rembert/__init__.py +++ b/src/transformers/models/rembert/__init__.py @@ -110,4 +110,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/retribert/__init__.py b/src/transformers/models/retribert/__init__.py index 4d07a49ea3..dc3f2f4ae6 100644 --- a/src/transformers/models/retribert/__init__.py +++ b/src/transformers/models/retribert/__init__.py @@ -54,4 +54,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/roberta/__init__.py b/src/transformers/models/roberta/__init__.py index 91058cf040..05c778f553 100644 --- a/src/transformers/models/roberta/__init__.py +++ b/src/transformers/models/roberta/__init__.py @@ -116,4 +116,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/roformer/__init__.py b/src/transformers/models/roformer/__init__.py index 85853a3c74..d7d8160e23 100644 --- a/src/transformers/models/roformer/__init__.py +++ b/src/transformers/models/roformer/__init__.py @@ -124,4 +124,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/segformer/__init__.py b/src/transformers/models/segformer/__init__.py index d0e8bfeee5..8c3ebc47e0 100644 --- a/src/transformers/models/segformer/__init__.py +++ b/src/transformers/models/segformer/__init__.py @@ -60,4 +60,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/sew/__init__.py b/src/transformers/models/sew/__init__.py index 1433d0dc06..239e8fbfc2 100644 --- a/src/transformers/models/sew/__init__.py +++ b/src/transformers/models/sew/__init__.py @@ -49,4 +49,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/sew_d/__init__.py b/src/transformers/models/sew_d/__init__.py index 90ed2889af..95b6a98fab 100644 --- a/src/transformers/models/sew_d/__init__.py +++ b/src/transformers/models/sew_d/__init__.py @@ -49,4 +49,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/speech_encoder_decoder/__init__.py b/src/transformers/models/speech_encoder_decoder/__init__.py index 01d552038a..5c8ad64455 100644 --- a/src/transformers/models/speech_encoder_decoder/__init__.py +++ b/src/transformers/models/speech_encoder_decoder/__init__.py @@ -37,4 +37,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/speech_to_text/__init__.py b/src/transformers/models/speech_to_text/__init__.py index 734bd4725b..c0cb8f9a81 100644 --- a/src/transformers/models/speech_to_text/__init__.py +++ b/src/transformers/models/speech_to_text/__init__.py @@ -68,4 +68,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/speech_to_text_2/__init__.py b/src/transformers/models/speech_to_text_2/__init__.py index 08ed595720..766035306e 100644 --- a/src/transformers/models/speech_to_text_2/__init__.py +++ b/src/transformers/models/speech_to_text_2/__init__.py @@ -53,4 +53,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/splinter/__init__.py b/src/transformers/models/splinter/__init__.py index fc755eb8ca..e2d8341db3 100644 --- a/src/transformers/models/splinter/__init__.py +++ b/src/transformers/models/splinter/__init__.py @@ -58,4 +58,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/squeezebert/__init__.py b/src/transformers/models/squeezebert/__init__.py index 5a1c91aa73..1abae4f538 100644 --- a/src/transformers/models/squeezebert/__init__.py +++ b/src/transformers/models/squeezebert/__init__.py @@ -66,4 +66,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/t5/__init__.py b/src/transformers/models/t5/__init__.py index 0b6e8f8ac4..7cb3cd8bb2 100644 --- a/src/transformers/models/t5/__init__.py +++ b/src/transformers/models/t5/__init__.py @@ -100,4 +100,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/tapas/__init__.py b/src/transformers/models/tapas/__init__.py index 9f1d442b98..3af8f8f90e 100644 --- a/src/transformers/models/tapas/__init__.py +++ b/src/transformers/models/tapas/__init__.py @@ -76,4 +76,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/transfo_xl/__init__.py b/src/transformers/models/transfo_xl/__init__.py index eb2ffcefa3..5e1ec08f1b 100644 --- a/src/transformers/models/transfo_xl/__init__.py +++ b/src/transformers/models/transfo_xl/__init__.py @@ -78,4 +78,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/trocr/__init__.py b/src/transformers/models/trocr/__init__.py index 4c2a60bc2f..de419b336b 100644 --- a/src/transformers/models/trocr/__init__.py +++ b/src/transformers/models/trocr/__init__.py @@ -47,4 +47,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/unispeech/__init__.py b/src/transformers/models/unispeech/__init__.py index c005d9c299..7df28d26d5 100644 --- a/src/transformers/models/unispeech/__init__.py +++ b/src/transformers/models/unispeech/__init__.py @@ -50,4 +50,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/unispeech_sat/__init__.py b/src/transformers/models/unispeech_sat/__init__.py index a6479e962d..b1c6a594aa 100644 --- a/src/transformers/models/unispeech_sat/__init__.py +++ b/src/transformers/models/unispeech_sat/__init__.py @@ -54,4 +54,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/vision_encoder_decoder/__init__.py b/src/transformers/models/vision_encoder_decoder/__init__.py index a30edb71d8..4f900c6cbc 100644 --- a/src/transformers/models/vision_encoder_decoder/__init__.py +++ b/src/transformers/models/vision_encoder_decoder/__init__.py @@ -49,4 +49,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/visual_bert/__init__.py b/src/transformers/models/visual_bert/__init__.py index b1f0c075d4..384914c458 100644 --- a/src/transformers/models/visual_bert/__init__.py +++ b/src/transformers/models/visual_bert/__init__.py @@ -58,4 +58,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/vit/__init__.py b/src/transformers/models/vit/__init__.py index a66edd8948..09b91776cd 100644 --- a/src/transformers/models/vit/__init__.py +++ b/src/transformers/models/vit/__init__.py @@ -73,4 +73,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/vit_mae/__init__.py b/src/transformers/models/vit_mae/__init__.py index e532e75551..7551c5f328 100644 --- a/src/transformers/models/vit_mae/__init__.py +++ b/src/transformers/models/vit_mae/__init__.py @@ -50,4 +50,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/wav2vec2/__init__.py b/src/transformers/models/wav2vec2/__init__.py index db27e5e0c4..24d5fb14db 100644 --- a/src/transformers/models/wav2vec2/__init__.py +++ b/src/transformers/models/wav2vec2/__init__.py @@ -97,4 +97,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/wav2vec2_phoneme/__init__.py b/src/transformers/models/wav2vec2_phoneme/__init__.py index 4ac68aa740..7d66bee633 100644 --- a/src/transformers/models/wav2vec2_phoneme/__init__.py +++ b/src/transformers/models/wav2vec2_phoneme/__init__.py @@ -32,4 +32,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/wav2vec2_with_lm/__init__.py b/src/transformers/models/wav2vec2_with_lm/__init__.py index cca731f0f7..d8fce39364 100644 --- a/src/transformers/models/wav2vec2_with_lm/__init__.py +++ b/src/transformers/models/wav2vec2_with_lm/__init__.py @@ -32,4 +32,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/wavlm/__init__.py b/src/transformers/models/wavlm/__init__.py index 2cfc854919..c9ee046d4e 100644 --- a/src/transformers/models/wavlm/__init__.py +++ b/src/transformers/models/wavlm/__init__.py @@ -52,4 +52,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/xlm/__init__.py b/src/transformers/models/xlm/__init__.py index b4479c5491..7de227e87a 100644 --- a/src/transformers/models/xlm/__init__.py +++ b/src/transformers/models/xlm/__init__.py @@ -86,4 +86,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/xlm_prophetnet/__init__.py b/src/transformers/models/xlm_prophetnet/__init__.py index d1937c9983..cd19330301 100644 --- a/src/transformers/models/xlm_prophetnet/__init__.py +++ b/src/transformers/models/xlm_prophetnet/__init__.py @@ -60,4 +60,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/xlm_roberta/__init__.py b/src/transformers/models/xlm_roberta/__init__.py index 7ef5dd2c03..26439a3051 100644 --- a/src/transformers/models/xlm_roberta/__init__.py +++ b/src/transformers/models/xlm_roberta/__init__.py @@ -104,4 +104,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/models/xlnet/__init__.py b/src/transformers/models/xlnet/__init__.py index 5a974cf9b6..22303316f3 100644 --- a/src/transformers/models/xlnet/__init__.py +++ b/src/transformers/models/xlnet/__init__.py @@ -104,4 +104,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/src/transformers/onnx/__init__.py b/src/transformers/onnx/__init__.py index 95200fa842..ddd8d9a2ab 100644 --- a/src/transformers/onnx/__init__.py +++ b/src/transformers/onnx/__init__.py @@ -47,4 +47,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py index 35970c7dad..323c3f2377 100644 --- a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py +++ b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py @@ -210,4 +210,4 @@ if TYPE_CHECKING: else: import sys - sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure) + sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) diff --git a/tests/test_configuration_auto.py b/tests/test_configuration_auto.py index a103b256c2..23b090dea7 100644 --- a/tests/test_configuration_auto.py +++ b/tests/test_configuration_auto.py @@ -13,10 +13,12 @@ # See the License for the specific language governing permissions and # limitations under the License. +import importlib import os import tempfile import unittest +import transformers.models.auto from transformers.models.auto.configuration_auto import CONFIG_MAPPING, AutoConfig from transformers.models.bert.configuration_bert import BertConfig from transformers.models.roberta.configuration_roberta import RobertaConfig @@ -31,6 +33,10 @@ class NewModelConfig(BertConfig): class AutoConfigTest(unittest.TestCase): + def test_module_spec(self): + self.assertIsNotNone(transformers.models.auto.__spec__) + self.assertIsNotNone(importlib.util.find_spec("transformers.models.auto")) + def test_config_from_model_shortcut(self): config = AutoConfig.from_pretrained("bert-base-uncased") self.assertIsInstance(config, BertConfig) diff --git a/tests/test_file_utils.py b/tests/test_file_utils.py index 4de449b344..4710f19d98 100644 --- a/tests/test_file_utils.py +++ b/tests/test_file_utils.py @@ -64,9 +64,11 @@ def context_fr(): print("Au revoir!") -def test_module_spec(): - assert transformers.__spec__ is not None - assert importlib.util.find_spec("transformers") is not None +class TestImportMechanisms(unittest.TestCase): + def test_module_spec_available(self): + # If the spec is missing, importlib would not be able to import the module dynamically. + assert transformers.__spec__ is not None + assert importlib.util.find_spec("transformers") is not None class GetFromCacheTests(unittest.TestCase):