replace assert with exception for padding_side arg in PreTrainedTokenizerBase __init__

This commit is contained in:
SaulLu 2022-02-01 11:12:37 +01:00
parent d12ae81664
commit 7a98b87962

View File

@ -1456,10 +1456,10 @@ class PreTrainedTokenizerBase(SpecialTokensMixin, PushToHubMixin):
# Padding side is right by default and overridden in subclasses. If specified in the kwargs, it is changed. # Padding side is right by default and overridden in subclasses. If specified in the kwargs, it is changed.
self.padding_side = kwargs.pop("padding_side", self.padding_side) self.padding_side = kwargs.pop("padding_side", self.padding_side)
assert self.padding_side in [ if self.padding_side not in ["right", "left"]:
"right", raise ValueError(
"left", f"Padding side should be selected between 'right' and 'left', current value: {self.padding_side}"
], f"Padding side should be selected between 'right' and 'left', current value: {self.padding_side}" )
self.model_input_names = kwargs.pop("model_input_names", self.model_input_names) self.model_input_names = kwargs.pop("model_input_names", self.model_input_names)
self.deprecation_warnings = ( self.deprecation_warnings = (