Skip to content

GitLab

Unverified Commit a2b1e1df authored by Noah Siegel's avatar Noah Siegel Committed by GitHub
Browse files

Fix typo in max_length deprecation warnings (#27788)

parent 7edf8bfa
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
src/transformers/generation/utils.py
View file @ a2b1e1df
...@@ -2801,7 +2801,7 @@ class GenerationMixin: ...@@ -2801,7 +2801,7 @@ class GenerationMixin:
if max_length is not None: if max_length is not None:
warnings.warn( warnings.warn(
"`max_length` is deprecated in this function, use" "`max_length` is deprecated in this function, use"
" `stopping_criteria=StoppingCriteriaList(MaxLengthCriteria(max_length=max_length))` instead.", " `stopping_criteria=StoppingCriteriaList([MaxLengthCriteria(max_length=max_length)])` instead.",
UserWarning, UserWarning,
) )
stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length) stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
...@@ -3080,7 +3080,7 @@ class GenerationMixin: ...@@ -3080,7 +3080,7 @@ class GenerationMixin:
if max_length is not None: if max_length is not None:
warnings.warn( warnings.warn(
"`max_length` is deprecated in this function, use" "`max_length` is deprecated in this function, use"
" `stopping_criteria=StoppingCriteriaList(MaxLengthCriteria(max_length=max_length))` instead.", " `stopping_criteria=StoppingCriteriaList([MaxLengthCriteria(max_length=max_length)])` instead.",
UserWarning, UserWarning,
) )
stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length) stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
...@@ -3422,7 +3422,7 @@ class GenerationMixin: ...@@ -3422,7 +3422,7 @@ class GenerationMixin:
if max_length is not None: if max_length is not None:
warnings.warn( warnings.warn(
"`max_length` is deprecated in this function, use" "`max_length` is deprecated in this function, use"
" `stopping_criteria=StoppingCriteriaList(MaxLengthCriteria(max_length=max_length))` instead.", " `stopping_criteria=StoppingCriteriaList([MaxLengthCriteria(max_length=max_length)])` instead.",
UserWarning, UserWarning,
) )
stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length) stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
...@@ -3750,7 +3750,7 @@ class GenerationMixin: ...@@ -3750,7 +3750,7 @@ class GenerationMixin:
if max_length is not None: if max_length is not None:
warnings.warn( warnings.warn(
"`max_length` is deprecated in this function, use" "`max_length` is deprecated in this function, use"
" `stopping_criteria=StoppingCriteriaList(MaxLengthCriteria(max_length=max_length))` instead.", " `stopping_criteria=StoppingCriteriaList([MaxLengthCriteria(max_length=max_length)])` instead.",
UserWarning, UserWarning,
) )
stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length) stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
...@@ -4140,7 +4140,7 @@ class GenerationMixin: ...@@ -4140,7 +4140,7 @@ class GenerationMixin:
if max_length is not None: if max_length is not None:
warnings.warn( warnings.warn(
"`max_length` is deprecated in this function, use" "`max_length` is deprecated in this function, use"
" `stopping_criteria=StoppingCriteriaList(MaxLengthCriteria(max_length=max_length))` instead.", " `stopping_criteria=StoppingCriteriaList([MaxLengthCriteria(max_length=max_length)])` instead.",
UserWarning, UserWarning,
) )
stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length) stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment