From 5fae56511b240350e5df59222c45df9b4faebc4c Mon Sep 17 00:00:00 2001 From: star-nox Date: Tue, 12 Dec 2023 18:45:21 -0600 Subject: [PATCH] minor function name change --- ai_ta_backend/filtering_contexts.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ai_ta_backend/filtering_contexts.py b/ai_ta_backend/filtering_contexts.py index a1eaf98c..cfc8c95b 100644 --- a/ai_ta_backend/filtering_contexts.py +++ b/ai_ta_backend/filtering_contexts.py @@ -209,7 +209,7 @@ def parse_result(result): # return 'yes' in line.lower() # return False -# def run(contexts, user_query, max_tokens_to_return=3000, max_time_before_return=None, max_concurrency=100): +# def ray_context_filtering(contexts, user_query, max_tokens_to_return=3000, max_time_before_return=None, max_concurrency=100): # # # Main function for filtering contexts using RAY. Use this when dealing with a list of contexts. # @@ -268,7 +268,7 @@ def parse_result(result): # start_time = time.monotonic() # # print(len(CONTEXTS)) -# final_passage_list = list(run(contexts=CONTEXTS*2, user_query=USER_QUERY, max_time_before_return=45, max_concurrency=20)) +# final_passage_list = list(ray_context_filtering(contexts=CONTEXTS*2, user_query=USER_QUERY, max_time_before_return=45, max_concurrency=20)) # print("✅✅✅ FINAL RESULTS: \n" + '\n'.join(json.dumps(r, indent=2) for r in final_passage_list)) # print("✅✅✅ TOTAL RETURNED: ", len(final_passage_list))