fix test_translation_performance

Ayush-iitkgp · Sep 11, 2024 · cdcf303 · cdcf303
1 parent 1eb8ef4
commit cdcf303
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmark/test_translation_performance.py b/tests/benchmark/test_translation_performance.py
@@ -50,7 +50,7 @@ def test_translation_performance(num_sections: int, num_paragraphs: int) -> None
     sync_duration = end_time - start_time
 
     start_time = time.time()
-    _ = HTMLTranslationService.translate_html(large_html)
+    _ = HTMLTranslationService.parallel_thread_translate_tags(tag)
     end_time = time.time()
     parallel_duration = end_time - start_time