From f58b211ed33e7e389cab63af1f8f208e8585df99 Mon Sep 17 00:00:00 2001 From: Ayaka Date: Mon, 24 Oct 2022 18:34:43 +0800 Subject: [PATCH] [Doctest] Add `configuration_electra.py` (#19807) --- src/transformers/models/electra/configuration_electra.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/electra/configuration_electra.py b/src/transformers/models/electra/configuration_electra.py index 3ea54aa7ca..01a6d1165a 100644 --- a/src/transformers/models/electra/configuration_electra.py +++ b/src/transformers/models/electra/configuration_electra.py @@ -119,12 +119,12 @@ class ElectraConfig(PretrainedConfig): Examples: ```python - >>> from transformers import ElectraModel, ElectraConfig + >>> from transformers import ElectraConfig, ElectraModel >>> # Initializing a ELECTRA electra-base-uncased style configuration >>> configuration = ElectraConfig() - >>> # Initializing a model from the electra-base-uncased style configuration + >>> # Initializing a model (with random weights) from the electra-base-uncased style configuration >>> model = ElectraModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 4b7275df8d..ded7172213 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -60,6 +60,7 @@ src/transformers/models/detr/configuration_detr.py src/transformers/models/detr/modeling_detr.py src/transformers/models/distilbert/configuration_distilbert.py src/transformers/models/dpt/modeling_dpt.py +src/transformers/models/electra/configuration_electra.py src/transformers/models/electra/modeling_electra.py src/transformers/models/electra/modeling_tf_electra.py src/transformers/models/ernie/configuration_ernie.py