fix: tests

huggingface · Jan 14, 2025 · 11533c5 · 11533c5
1 parent 3fc5e55
commit 11533c5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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."),
-        }
+        };
     }
 }