Update utils/query_element_selection.py

utils/query_element_selection.py

@@ -97,7 +97,7 @@ def easy_level_element_selection(day_list):
 
 
 
-def middle_level_element_selection(day_list):
+def medium_level_element_selection(day_list):
     days = random.choice(day_list)
     date  = [date.strftime('%Y-%m-%d') for date in select_consecutive_dates(days)]
     people_number = random.choice(random.choice([[2],[3,4,5,6,7,8]]))
@@ -141,7 +141,7 @@ def middle_level_element_selection(day_list):
     elif days == 7:
         final_budget = round_to_hundreds(round_to_hundreds((budget["average"]+budget["highest"])/2) * people_number * 0.75)
 
-    query_dict = {"org": final_org, "dest": final_des, "days": days, "visiting_city_number":visiting_city_map[days], "date":date, "people_number": people_number, "local_constraint": local_constrain_record ,"budget": final_budget,"query": None, "level":"middle"}
+    query_dict = {"org": final_org, "dest": final_des, "days": days, "visiting_city_number":visiting_city_map[days], "date":date, "people_number": people_number, "local_constraint": local_constrain_record ,"budget": final_budget,"query": None, "level":"medium"}
     return query_dict
 
 
@@ -212,8 +212,8 @@ def generate_elements(number:int, level="easy", day_list=[3,5,7]):
                 query = easy_level_element_selection(day_list)
                 if query not in query_list:
                     query_list.append(query)
-            elif level == "middle":
-                query = middle_level_element_selection(day_list)
+            elif level == "medium":
+                query = medium_level_element_selection(day_list)
                 if query not in query_list:
                     query_list.append(query)
             elif level == "hard":
@@ -229,9 +229,9 @@ def main():
 
     # save query_list as jsonl file
     for num, day_list in zip([160,160,160], [[3],[5],[7]]):
-        query_list = generate_elements(num,"middle",day_list=day_list)
+        query_list = generate_elements(num,"medium",day_list=day_list)
 
-        with open('../data/query/final_annotation_middle.jsonl', 'a+') as f:
+        with open('../data/query/final_annotation_medium.jsonl', 'a+') as f:
             for query in query_list:
                 # print(query)
                 json.dump(query, f)