From 1d4d9dc3c9ff145f31a7c241f993a8f559f1a23e Mon Sep 17 00:00:00 2001 From: RamitPahwa Date: Thu, 13 Oct 2022 04:56:35 -0500 Subject: [PATCH] [Doctest] Fixing mobile bert configuration doctest (#19557) * Fixing mobile bert configuration doctest * Fixed build failures by removing empty line --- .../models/mobilebert/configuration_mobilebert.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/mobilebert/configuration_mobilebert.py b/src/transformers/models/mobilebert/configuration_mobilebert.py index 73b8844ed7..7034cdb276 100644 --- a/src/transformers/models/mobilebert/configuration_mobilebert.py +++ b/src/transformers/models/mobilebert/configuration_mobilebert.py @@ -93,12 +93,12 @@ class MobileBertConfig(PretrainedConfig): Examples: ```python - >>> from transformers import MobileBertModel, MobileBertConfig + >>> from transformers import MobileBertConfig, MobileBertModel >>> # Initializing a MobileBERT configuration >>> configuration = MobileBertConfig() - >>> # Initializing a model from the configuration above + >>> # Initializing a model (with random weights) from the configuration above >>> model = MobileBertModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 2024f9d235..8460bf1e8e 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -61,6 +61,7 @@ src/transformers/models/longt5/modeling_longt5.py src/transformers/models/marian/modeling_marian.py src/transformers/models/markuplm/modeling_markuplm.py src/transformers/models/mbart/modeling_mbart.py +src/transformers/models/mobilebert/configuration_mobilebert.py src/transformers/models/mobilebert/modeling_mobilebert.py src/transformers/models/mobilebert/modeling_tf_mobilebert.py src/transformers/models/mobilevit/modeling_mobilevit.py