Adding __repr__ function and repr=True in dataclass

This commit is contained in:
Uri Alon 2023-09-19 01:21:31 +00:00
parent 1a1439225a
commit 18839505d1

View File

@ -78,7 +78,7 @@ if is_tokenizers_available():
from tokenizers import Encoding as EncodingFast
else:
@dataclass(frozen=False, eq=True)
@dataclass(repr=True, frozen=False, eq=True)
class AddedToken:
"""
AddedToken represents a token to be added to a Tokenizer An AddedToken can have special options defining the
@ -104,6 +104,9 @@ else:
def __str__(self):
return self.content
def __repr__(self):
return self.content
@dataclass
class EncodingFast:
"""This is dummy class because without the `tokenizers` library we don't have these objects anyway"""