Python - Add encoding getters

This commit is contained in:
Anthony MOI
2019-12-10 15:17:41 -05:00
parent 8cedc5f1f6
commit 99773d9ce4

View File

@ -13,3 +13,36 @@ impl Encoding {
Encoding { encoding }
}
}
#[pymethods]
impl Encoding {
#[getter]
fn get_original(&self) -> String {
self.encoding.get_original().to_owned()
}
#[getter]
fn get_normalized(&self) -> String {
self.encoding.get_normalized().to_owned()
}
#[getter]
fn get_ids(&self) -> Vec<u32> {
self.encoding.get_ids().to_vec()
}
#[getter]
fn get_tokens(&self) -> Vec<String> {
self.encoding.get_tokens().to_vec()
}
#[getter]
fn get_type_ids(&self) -> Vec<u32> {
self.encoding.get_type_ids().to_vec()
}
#[getter]
fn get_offsets(&self) -> Vec<(usize, usize)> {
self.encoding.get_offsets().to_vec()
}
}