Skip to content

Commit

Permalink
minor function name change
Browse files Browse the repository at this point in the history
  • Loading branch information
star-nox committed Dec 13, 2023
1 parent 4542f15 commit 5fae565
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ai_ta_backend/filtering_contexts.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
#
Expand Down Expand Up @@ -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))
Expand Down

0 comments on commit 5fae565

Please sign in to comment.