From 2d97bc40e6a13ea7b45afaa7e837054dbaefa4c5 Mon Sep 17 00:00:00 2001 From: kagami Date: Sun, 19 May 2024 16:50:30 +0800 Subject: [PATCH] Skip 2 llm calls temporarily --- merico/unit_tests/ut_workflow.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/merico/unit_tests/ut_workflow.py b/merico/unit_tests/ut_workflow.py index b4c650b..8a8e719 100644 --- a/merico/unit_tests/ut_workflow.py +++ b/merico/unit_tests/ut_workflow.py @@ -6,7 +6,6 @@ from find_context import ( Position, Range, find_symbol_context_by_static_analysis, - find_symbol_context_of_llm_recommendation, ) from find_reference_tests import find_reference_tests from i18n import TUILanguage, get_translation @@ -52,8 +51,10 @@ class UnitTestsWorkflow: contexts.update(v) contexts = list(contexts) - print("- Finding reference files...", flush=True) - reference_files = self.step_2_find_reference_files() + reference_files = [] + # TODO: Restore this feature after checking the performance issue + # print("- Finding reference files...", flush=True) + # reference_files = self.step_2_find_reference_files() print("- Proposing test cases...", flush=True) cases = self.step_3_propose_cases(contexts) @@ -88,11 +89,12 @@ class UnitTestsWorkflow: {item for sublist in list(symbol_context.values()) for item in sublist} ) - recommended_context = find_symbol_context_of_llm_recommendation( - self.func_to_test, known_context_for_llm - ) + # TODO: Restore this feature after checking the performance issue + # recommended_context = find_symbol_context_of_llm_recommendation( + # self.func_to_test, known_context_for_llm + # ) - symbol_context.update(recommended_context) + # symbol_context.update(recommended_context) return symbol_context