diff --git a/bindings/python/src/processors.rs b/bindings/python/src/processors.rs index a8dd31486..945d8b2fc 100644 --- a/bindings/python/src/processors.rs +++ b/bindings/python/src/processors.rs @@ -602,7 +602,7 @@ pub fn processors(m: &Bound<'_, PyModule>) -> PyResult<()> { #[cfg(test)] mod test { - use std::sync::Arc; + use std::sync::{Arc, RwLock}; use pyo3::prelude::*; use tk::processors::bert::BertProcessing; @@ -637,15 +637,15 @@ mod test { assert_eq!(py_ser, rs_wrapper_ser); let py_processing: PyPostProcessor = serde_json::from_str(&rs_processing_ser).unwrap(); - match py_processing.processor.as_ref() { + match *py_processing.processor.as_ref().read().unwrap() { PostProcessorWrapper::Bert(_) => (), _ => panic!("Expected Bert postprocessor."), } let py_processing: PyPostProcessor = serde_json::from_str(&rs_wrapper_ser).unwrap(); - match py_processing.processor.as_ref() { + match *py_processing.processor.as_ref().read().unwrap() { PostProcessorWrapper::Bert(_) => (), _ => panic!("Expected Bert postprocessor."), - } + }; } }