Skip to content

Commit a26f0fa

Browse files
authored
Fix auto_docstring crashing when dependencies are missing (#39564)
* add try except to not crash auto_docstring when some dependency are missing * safeguard None value in placeholder dict
1 parent 69cff31 commit a26f0fa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/utils/auto_docstring.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1154,7 +1154,7 @@ def get_placeholders_dict(placeholders: list, model_name: str) -> dict:
11541154
if place_holder_value is not None:
11551155
if isinstance(place_holder_value, (list, tuple)):
11561156
place_holder_value = place_holder_value[0]
1157-
placeholders_dict[placeholder] = place_holder_value
1157+
placeholders_dict[placeholder] = place_holder_value if place_holder_value is not None else placeholder
11581158
else:
11591159
placeholders_dict[placeholder] = placeholder
11601160

0 commit comments

Comments
 (0)