mirror of
https://github.com/mii443/tokenizers.git
synced 2025-08-22 16:25:30 +00:00
make special editable as well
This commit is contained in:
@ -216,6 +216,12 @@ impl PyAddedToken {
|
||||
self.get_token().special
|
||||
}
|
||||
|
||||
/// Set the value of the :obj:`special` option
|
||||
#[setter]
|
||||
fn set_special(&mut self, special: bool) {
|
||||
self.special = special;
|
||||
}
|
||||
|
||||
fn __str__(&self) -> PyResult<&str> {
|
||||
Ok(&self.content)
|
||||
}
|
||||
|
@ -20,6 +20,10 @@ class TestAddedToken:
|
||||
assert added_token.content == "<MASK>"
|
||||
assert type(added_token) == AddedToken
|
||||
added_token.content = added_token.content.lower()
|
||||
|
||||
assert added_token.special == False
|
||||
added_token.special = True
|
||||
assert added_token.special == True
|
||||
|
||||
assert str(added_token) == "<mask>"
|
||||
assert (
|
||||
|
@ -678,6 +678,10 @@ mod tests {
|
||||
let mut token: AddedToken = AddedToken::from("Hey", false);
|
||||
token.content = "hey".to_string();
|
||||
assert_eq!(token.content, "hey"); // Token was already there
|
||||
|
||||
token.special = true;
|
||||
assert_eq!(token.special, true); // Token was already there
|
||||
|
||||
}
|
||||
|
||||
#[test]
|
||||
|
Reference in New Issue
Block a user