From 5e012f8e3c3a8d8bb5d6d95c58ff59a4a3156519 Mon Sep 17 00:00:00 2001 From: Saad Mahmud Date: Wed, 16 Nov 2022 23:20:06 +0600 Subject: [PATCH] [Doctest] Add configuration_deformable_detr.py (#20273) * Update configuration_deformable_detr.py comment * Add DeformableDetrConfig to documentation_tests.txt --- .../models/deformable_detr/configuration_deformable_detr.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/deformable_detr/configuration_deformable_detr.py b/src/transformers/models/deformable_detr/configuration_deformable_detr.py index b5900b316ffa86..218f6d3506e921 100644 --- a/src/transformers/models/deformable_detr/configuration_deformable_detr.py +++ b/src/transformers/models/deformable_detr/configuration_deformable_detr.py @@ -120,12 +120,12 @@ class DeformableDetrConfig(PretrainedConfig): Examples: ```python - >>> from transformers import DeformableDetrModel, DeformableDetrConfig + >>> from transformers import DeformableDetrConfig, DeformableDetrModel >>> # Initializing a Deformable DETR SenseTime/deformable-detr style configuration >>> configuration = DeformableDetrConfig() - >>> # Initializing a model from the SenseTime/deformable-detr style configuration + >>> # Initializing a model (with random weights) from the SenseTime/deformable-detr style configuration >>> model = DeformableDetrModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 6b0e33cc2748fa..200c836860f6c1 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -58,6 +58,7 @@ src/transformers/models/deberta/modeling_deberta.py src/transformers/models/deberta_v2/configuration_deberta_v2.py src/transformers/models/deberta_v2/modeling_deberta_v2.py src/transformers/models/decision_transformer/configuration_decision_transformer.py +src/transformers/models/deformable_detr/configuration_deformable_detr.py src/transformers/models/deformable_detr/modeling_deformable_detr.py src/transformers/models/deit/configuration_deit.py src/transformers/models/deit/modeling_deit.py