From 3d320c78c32334f66d72d57ff6322d9e3a7dc00b Mon Sep 17 00:00:00 2001
From: Sujay <89520981+SD-13@users.noreply.github.com>
Date: Thu, 13 Oct 2022 22:37:10 +0530
Subject: [PATCH] [Doctest] adds trajectory_transformer config to Docs test
 (#19586)

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

diff --git a/src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py b/src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py
index 537a467c70..fcfc2b1f33 100644
--- a/src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py
+++ b/src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py
@@ -93,12 +93,12 @@ class TrajectoryTransformerConfig(PretrainedConfig):
         Example:
 
     ```python
-    >>> from transformers import TrajectoryTransformerModel, TrajectoryTransformerConfig
+    >>> from transformers import TrajectoryTransformerConfig, TrajectoryTransformerModel
 
     >>> # Initializing a TrajectoryTransformer CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
     >>> configuration = TrajectoryTransformerConfig()
 
-    >>> # Initializing a model from the CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
+    >>> # Initializing a model (with random weights) from the CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
     >>> model = TrajectoryTransformerModel(configuration)
 
     >>> # Accessing the model configuration
diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt
index 9d90713b9b..ec1093510e 100644
--- a/utils/documentation_tests.txt
+++ b/utils/documentation_tests.txt
@@ -96,6 +96,7 @@ src/transformers/models/speech_to_text/modeling_speech_to_text.py
 src/transformers/models/speech_to_text_2/modeling_speech_to_text_2.py
 src/transformers/models/segformer/modeling_tf_segformer.py
 src/transformers/models/swin/modeling_swin.py
+src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py
 src/transformers/models/trocr/modeling_trocr.py
 src/transformers/models/unispeech/modeling_unispeech.py
 src/transformers/models/unispeech_sat/modeling_unispeech_sat.py