From 909f07092a046908e727afa8bca71062e1025bf9 Mon Sep 17 00:00:00 2001
From: Xabier Lahuerta Vazquez <xlahuerta@protonmail.com>
Date: Fri, 14 Oct 2022 15:17:36 +0200
Subject: [PATCH] [Doctest] Add `configuration_bigbird_pegasus.py` and
 `configuration_big_bird.py` (#19606)

* [Doctest] Add `configuration_bigbird_pegasus.py` and `configuration_big_bird`

[Doctest] Re-style `configuration_big_bird.py`

* [Doctest] One python instruction per line

* [Doctest] Fix styling

* [Doctest] More styling fixes
---
 .../models/big_bird/configuration_big_bird.py  | 18 ++++++++----------
 .../configuration_bigbird_pegasus.py           | 18 ++++++++----------
 utils/documentation_tests.txt                  |  2 ++
 3 files changed, 18 insertions(+), 20 deletions(-)

diff --git a/src/transformers/models/big_bird/configuration_big_bird.py b/src/transformers/models/big_bird/configuration_big_bird.py
index 371846982f..ed309bbf8c 100644
--- a/src/transformers/models/big_bird/configuration_big_bird.py
+++ b/src/transformers/models/big_bird/configuration_big_bird.py
@@ -91,19 +91,17 @@ class BigBirdConfig(PretrainedConfig):
     Example:
 
     ```python
+    >>> from transformers import BigBirdConfig, BigBirdModel
 
-    ```
+    >>> # Initializing a BigBird google/bigbird-roberta-base style configuration
+    >>> configuration = BigBirdConfig()
 
-        >>> from transformers import BigBirdModel, BigBirdConfig
+    >>> # Initializing a model (with random weights) from the google/bigbird-roberta-base style configuration
+    >>> model = BigBirdModel(configuration)
 
-        >>> # Initializing a BigBird google/bigbird-roberta-base style configuration >>> configuration =
-        BigBirdConfig()
-
-        >>> # Initializing a model from the google/bigbird-roberta-base style configuration >>> model =
-        BigBirdModel(configuration)
-
-        >>> # Accessing the model configuration >>> configuration = model.config
-    """
+    >>> # Accessing the model configuration
+    >>> configuration = model.config
+    ```"""
     model_type = "big_bird"
 
     def __init__(
diff --git a/src/transformers/models/bigbird_pegasus/configuration_bigbird_pegasus.py b/src/transformers/models/bigbird_pegasus/configuration_bigbird_pegasus.py
index f0c874f694..ed9c0a42e0 100644
--- a/src/transformers/models/bigbird_pegasus/configuration_bigbird_pegasus.py
+++ b/src/transformers/models/bigbird_pegasus/configuration_bigbird_pegasus.py
@@ -109,19 +109,17 @@ class BigBirdPegasusConfig(PretrainedConfig):
     Example:
 
     ```python
+    >>> from transformers import BigBirdPegasusConfig, BigBirdPegasusModel
 
-    ```
+    >>> # Initializing a BigBirdPegasus bigbird-pegasus-base style configuration
+    >>> configuration = BigBirdPegasusConfig()
 
-        >>> from transformers import BigBirdPegasusModel, BigBirdPegasusConfig
+    >>> # Initializing a model (with random weights) from the bigbird-pegasus-base style configuration
+    >>> model = BigBirdPegasusModel(configuration)
 
-        >>> # Initializing a BigBirdPegasus bigbird-pegasus-base style configuration >>> configuration =
-        BigBirdPegasusConfig()
-
-        >>> # Initializing a model from the bigbird-pegasus-base style configuration >>> model =
-        BigBirdPegasusModel(configuration)
-
-        >>> # Accessing the model configuration >>> configuration = model.config
-    """
+    >>> # Accessing the model configuration
+    >>> configuration = model.config
+    ```"""
     model_type = "bigbird_pegasus"
     keys_to_ignore_at_inference = ["past_key_values"]
     attribute_map = {
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index 1b9b5c5b08..c53f9ac4f0 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -23,7 +23,9 @@ src/transformers/models/bert/configuration_bert.py
 src/transformers/models/bert/modeling_bert.py
 src/transformers/models/bert/modeling_tf_bert.py
 src/transformers/models/bert_generation/configuration_bert_generation.py
+src/transformers/models/bigbird_pegasus/configuration_bigbird_pegasus.py
 src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py
+src/transformers/models/big_bird/configuration_big_bird.py
 src/transformers/models/big_bird/modeling_big_bird.py
 src/transformers/models/blenderbot/configuration_blenderbot.py
 src/transformers/models/blenderbot/modeling_blenderbot.py