From 65d36ee8610b57cd27864898e820300b9cf75746 Mon Sep 17 00:00:00 2001
From: Xabier Lahuerta Vazquez <xlahuerta@protonmail.com>
Date: Wed, 19 Oct 2022 18:12:34 +0200
Subject: [PATCH] [Doctest] Add `configuration_decision_transformer.py`
 (#19751)

---
 .../configuration_decision_transformer.py                   | 6 +++---
 utils/documentation_tests.txt                               | 1 +
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/transformers/models/decision_transformer/configuration_decision_transformer.py b/src/transformers/models/decision_transformer/configuration_decision_transformer.py
index 01c74c247b..fc11b10624 100644
--- a/src/transformers/models/decision_transformer/configuration_decision_transformer.py
+++ b/src/transformers/models/decision_transformer/configuration_decision_transformer.py
@@ -90,13 +90,13 @@ class DecisionTransformerConfig(PretrainedConfig):
     Example:
 
     ```python
-    >>> from transformers import DecisionTransformerModel, DecisionTransformerConfig
+    >>> from transformers import DecisionTransformerConfig, DecisionTransformerModel
 
     >>> # Initializing a DecisionTransformer configuration
     >>> configuration = DecisionTransformerConfig()
 
-    >>> # Initializing a model from the configuration
-    >>> model = DecisionTransformerConfig(configuration)
+    >>> # Initializing a model (with random weights) from the configuration
+    >>> model = DecisionTransformerModel(configuration)
 
     >>> # Accessing the model configuration
     >>> configuration = model.config
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index e21463efe0..c42b0cb9ec 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -51,6 +51,7 @@ src/transformers/models/data2vec/modeling_data2vec_audio.py
 src/transformers/models/data2vec/modeling_data2vec_vision.py
 src/transformers/models/deberta/modeling_deberta.py
 src/transformers/models/deberta_v2/modeling_deberta_v2.py
+src/transformers/models/decision_transformer/configuration_decision_transformer.py
 src/transformers/models/deformable_detr/modeling_deformable_detr.py
 src/transformers/models/deit/configuration_deit.py
 src/transformers/models/deit/modeling_deit.py