From 835b0a05330f1ab5f9b18235d11d081504fc6425 Mon Sep 17 00:00:00 2001
From: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>
Date: Mon, 18 Sep 2023 18:58:01 +0100
Subject: [PATCH] [Check] Fix config docstring (#26222)

---
 utils/check_config_docstrings.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/utils/check_config_docstrings.py b/utils/check_config_docstrings.py
index 0a0f0a26ac..02ec510bab 100644
--- a/utils/check_config_docstrings.py
+++ b/utils/check_config_docstrings.py
@@ -16,7 +16,6 @@
 import inspect
 import re
 
-from transformers import CTRLConfig
 from transformers.utils import direct_transformers_import
 
 
@@ -74,9 +73,7 @@ def get_checkpoint_from_config_class(config_class):
 def check_config_docstrings_have_checkpoints():
     configs_without_checkpoint = []
 
-    a = [CTRLConfig]
-
-    for config_class in a:
+    for config_class in list(CONFIG_MAPPING.values()):
         # Skip deprecated models
         if "models.deprecated" in config_class.__module__:
             continue