mii/tokenizers
1
0
Fork 0
mirror of https://github.com/mii443/tokenizers.git synced 2025-12-06 04:38:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Python - Expose PostProcessors

Browse Source
This commit is contained in:
Anthony MOI
2019-12-16 18:46:14 -05:00
parent 1a90cc96e5
commit 93a74aa53a
7 changed files with 110 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bindings/python/src/lib.rs
View File

@@ -3,6 +3,7 @@ mod encoding;
mod error;
mod models;
mod pre_tokenizers;
mod processors;
mod token;
mod tokenizer;
mod trainers;
@@ -46,6 +47,14 @@ fn decoders(_py: Python, m: &PyModule) -> PyResult<()> {
Ok(())
}
/// Processors Module
#[pymodule]
fn processors(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_class::<processors::PostProcessor>()?;
m.add_class::<processors::BertProcessing>()?;
Ok(())
}
/// Tokenizers Module
#[pymodule]
fn tokenizers(_py: Python, m: &PyModule) -> PyResult<()> {
@@ -53,6 +62,7 @@ fn tokenizers(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_wrapped(wrap_pymodule!(models))?;
m.add_wrapped(wrap_pymodule!(pre_tokenizers))?;
m.add_wrapped(wrap_pymodule!(decoders))?;
m.add_wrapped(wrap_pymodule!(processors))?;
m.add_wrapped(wrap_pymodule!(trainers))?;
Ok(())
}