Python - Add some missing Encoding bindings

2025-08-23 00:35:35 +00:00 · 2019-12-16 19:38:18 -05:00
parent cc9f9107fa
commit d80f752ec9
2 changed files with 20 additions and 1 deletions
--- a/bindings/python/src/encoding.rs
+++ b/bindings/python/src/encoding.rs
@ -56,4 +56,19 @@ impl Encoding {
    fn get_offsets(&self) -> Vec<(usize, usize)> {
        self.encoding.get_offsets().to_vec()
    }
+
+    #[getter]
+    fn get_special_tokens_mask(&self) -> Vec<u32> {
+        self.encoding.get_special_tokens_mask().to_vec()
+    }
+
+    #[getter]
+    fn get_attention_mask(&self) -> Vec<u32> {
+        self.encoding.get_attention_mask().to_vec()
+    }
+
+    #[getter]
+    fn get_overflowing(&self) -> Option<Encoding> {
+        self.encoding.get_overflowing().cloned().map(Encoding::new)
+    }
 }