fix test & benchmark

mediar-ai · Aug 24, 2024 · c028387 · louis030195 · Aug 24, 2024 · c028387
1 parent b175a64
commit c028387
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/screenpipe-vision/benches/ocr_benchmark.rs b/screenpipe-vision/benches/ocr_benchmark.rs
@@ -169,7 +169,7 @@ fn bench_windows_ocr(c: &mut Criterion) {
 
                     for _ in 0..iters {
                         let start = std::time::Instant::now();
-                        let (result, _) = perform_ocr_windows(black_box(&image)).await;
+                        let (result, _, _) = perform_ocr_windows(black_box(&image)).await;
                         total_duration += start.elapsed();
 
                         let accuracy = calculate_accuracy(&result, EXPECTED_KEYWORDS);

diff --git a/screenpipe-vision/tests/integration_test.rs b/screenpipe-vision/tests/integration_test.rs
@@ -17,7 +17,7 @@ fn test_ocr_output() -> Result<(), Box<dyn std::error::Error>> {
     // Start timing
     let start = Instant::now();
 
-    let (text, json_output) = perform_ocr_tesseract(&image);
+    let (text, json_output, _) = perform_ocr_tesseract(&image);
 
     // Stop timing
     let duration = start.elapsed();