From 51d21b761967b395b4da72c08ca58396dc2f9e9e Mon Sep 17 00:00:00 2001
From: Partho <parthodas6176@gmail.com>
Date: Wed, 12 Oct 2022 23:31:25 +0530
Subject: [PATCH] [Doctest] Add `configuration_yolos.py` (#19539)

* YOLOS Config for doctest

* fix
---
 src/transformers/models/yolos/configuration_yolos.py | 4 ++--
 utils/documentation_tests.txt                        | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/transformers/models/yolos/configuration_yolos.py b/src/transformers/models/yolos/configuration_yolos.py
index 179d2833a1..c6bfbff444 100644
--- a/src/transformers/models/yolos/configuration_yolos.py
+++ b/src/transformers/models/yolos/configuration_yolos.py
@@ -92,12 +92,12 @@ class YolosConfig(PretrainedConfig):
     Example:
 
     ```python
-    >>> from transformers import YolosModel, YolosConfig
+    >>> from transformers import YolosConfig, YolosModel
 
     >>> # Initializing a YOLOS hustvl/yolos-base style configuration
     >>> configuration = YolosConfig()
 
-    >>> # Initializing a model from the hustvl/yolos-base style configuration
+    >>> # Initializing a model (with random weights) from the hustvl/yolos-base style configuration
     >>> model = YolosModel(configuration)
 
     >>> # Accessing the model configuration
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index 49107cd6f8..e87b8746c7 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -101,5 +101,6 @@ src/transformers/models/wav2vec2_with_lm/processing_wav2vec2_with_lm.py
 src/transformers/models/wavlm/modeling_wavlm.py
 src/transformers/models/whisper/modeling_whisper.py
 src/transformers/models/whisper/modeling_tf_whisper.py
+src/transformers/models/yolos/configuration_yolos.py
 src/transformers/models/yolos/modeling_yolos.py
 src/transformers/models/x_clip/modeling_x_clip.py