Fix doctests for audio models (#20468)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

Fix doctests for audio models (#20468)
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
f244a978 · Yih-Dar · GitHub · df938fc1 · f244a978
Unverified Commit f244a978 authored Nov 28, 2022 by Yih-Dar Committed by GitHub Nov 28, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/utils/doc.py src/transformers/utils/doc.py +1 -1

No files found.
--- a/src/transformers/utils/doc.py
+++ b/src/transformers/utils/doc.py
@@ -1087,7 +1087,7 @@ def add_code_sample_docstrings(
            expected_loss=expected_loss,
        )

-        if ["SequenceClassification" in model_class or "AudioClassification" in model_class] and modality == "audio":
+        if ("SequenceClassification" in model_class or "AudioClassification" in model_class) and modality == "audio":
            code_sample = sample_docstrings["AudioClassification"]
        elif "SequenceClassification" in model_class:
            code_sample = sample_docstrings["SequenceClassification"]