From dd464e22a756b96e3c20e459139da407798ab282 Mon Sep 17 00:00:00 2001
From: AymenBer99 <67442508+AymenBer99@users.noreply.github.com>
Date: Sat, 15 Oct 2022 12:35:35 +0200
Subject: [PATCH] [Doctest] CodeGen config for doctest (#19633)

---
 src/transformers/models/codegen/configuration_codegen.py | 4 ++--
 utils/documentation_tests.txt                            | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/transformers/models/codegen/configuration_codegen.py b/src/transformers/models/codegen/configuration_codegen.py
index 427715c622..292188e1ec 100644
--- a/src/transformers/models/codegen/configuration_codegen.py
+++ b/src/transformers/models/codegen/configuration_codegen.py
@@ -87,12 +87,12 @@ class CodeGenConfig(PretrainedConfig):
     Example:
 
     ```python
-    >>> from transformers import CodeGenModel, CodeGenConfig
+    >>> from transformers import CodeGenConfig, CodeGenModel
 
     >>> # Initializing a CodeGen 6B configuration
     >>> configuration = CodeGenConfig()
 
-    >>> # Initializing a model from the configuration
+    >>> # Initializing a model (with random weights) from the configuration
     >>> model = CodeGenModel(configuration)
 
     >>> # Accessing the model configuration
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index a077f698a0..abd3a77564 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -33,6 +33,7 @@ src/transformers/models/blenderbot_small/configuration_blenderbot_small.py
 src/transformers/models/blenderbot_small/modeling_blenderbot_small.py
 src/transformers/models/bloom/configuration_bloom.py
 src/transformers/models/canine/configuration_canine.py
+src/transformers/models/codegen/configuration_codegen.py
 src/transformers/models/conditional_detr/modeling_conditional_detr.py
 src/transformers/models/convnext/modeling_convnext.py
 src/transformers/models/ctrl/configuration_ctrl.py