[Doctest] Add configuration_deformable_detr.py (huggingface#20273)

* Update configuration_deformable_detr.py comment * Add DeformableDetrConfig to documentation_tests.txt
mpierrau · Dec 15, 2022 · af025b0 · af025b0
1 parent 64f0573
commit af025b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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