Using serde (serde_pyo3) to get __str__ and __repr__ easily.

This commit is contained in:
Nicolas Patry
2024-08-02 18:41:54 +02:00
parent 7415e28536
commit 86138337fc
8 changed files with 27 additions and 7 deletions

View File

@ -16,8 +16,8 @@ use tokenizers as tk;
/// Trainer will return an instance of this class when instantiated.
#[pyclass(module = "tokenizers.trainers", name = "Trainer", subclass)]
#[derive(Clone, Deserialize, Serialize)]
#[serde(transparent)]
pub struct PyTrainer {
#[serde(flatten)]
pub trainer: Arc<RwLock<TrainerWrapper>>,
}