From 2719599a22cf756e692fdc1eb6a5c9ae257840b2 Mon Sep 17 00:00:00 2001 From: Partho Date: Thu, 13 Oct 2022 15:33:15 +0530 Subject: [PATCH] [Doctest] Reformer Config for doctest (#19562) --- src/transformers/models/reformer/configuration_reformer.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/reformer/configuration_reformer.py b/src/transformers/models/reformer/configuration_reformer.py index ea2a1abd08..d4ffb2a39b 100755 --- a/src/transformers/models/reformer/configuration_reformer.py +++ b/src/transformers/models/reformer/configuration_reformer.py @@ -146,12 +146,12 @@ class ReformerConfig(PretrainedConfig): Examples: ```python - >>> from transformers import ReformerModel, ReformerConfig + >>> from transformers import ReformerConfig, ReformerModel >>> # Initializing a Reformer configuration >>> configuration = ReformerConfig() - >>> # Initializing a Reformer model + >>> # Initializing a Reformer model (with random weights) >>> model = ReformerModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 63fe29d6db..9864a284b5 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -75,6 +75,7 @@ src/transformers/models/pegasus/modeling_pegasus.py src/transformers/models/perceiver/modeling_perceiver.py src/transformers/models/plbart/modeling_plbart.py src/transformers/models/poolformer/modeling_poolformer.py +src/transformers/models/reformer/configuration_reformer.py src/transformers/models/reformer/modeling_reformer.py src/transformers/models/regnet/modeling_regnet.py src/transformers/models/regnet/modeling_tf_regnet.py