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 - Add NormalizedString + doc/typings

Browse Source
This commit is contained in:
Anthony MOI
2020-01-06 17:55:22 -05:00
parent 6de04bbaea
commit 0079a7a6b7
4 changed files with 93 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,6 +2,7 @@ mod decoders;
mod encoding;
mod error;
mod models;
mod normalized_string;
mod normalizers;
mod pre_tokenizers;
mod processors;
@@ -70,6 +71,8 @@ fn normalizers(_py: Python, m: &PyModule) -> PyResult<()> {
#[pymodule]
fn tokenizers(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_class::<tokenizer::Tokenizer>()?;
m.add_class::<encoding::Encoding>()?;
m.add_class::<normalized_string::NormalizedString>()?;
m.add_wrapped(wrap_pymodule!(models))?;
m.add_wrapped(wrap_pymodule!(pre_tokenizers))?;
m.add_wrapped(wrap_pymodule!(decoders))?;