step2 prepare_preference_dataset fix bug

step2_prepare_preference_dataset.py

@@ -60,7 +60,7 @@ def calculate_edit_distances(problems):
     return list(chain.from_iterable(results))
 
 
-def edit_distance(all_pairs, k, n, is_max=True):
+def mk_edit_distance_dataset(all_pairs, k, n, is_max=True):
     """
     Top-k pairs with the maximum/minimum edit distance.
     Each problem can contribute no more than n pairs.
@@ -96,7 +96,7 @@ if __name__ == "__main__":
     all_edit_distance_pairs = calculate_edit_distances(problems)
 
     # Maximum distance
-    preference_pairs, metadata = edit_distance(
+    preference_pairs, metadata = mk_edit_distance_dataset(
         all_edit_distance_pairs, 10 * 1000, 5, is_max=True
     )
     dataset_info = mk_dataset_info("apps_max_edit_distance_prefrence")
@@ -113,7 +113,7 @@ if __name__ == "__main__":
     )
 
     # Minimum distance
-    preference_pairs, metadata = edit_distance(
+    preference_pairs, metadata = mk_edit_distance_dataset(
         all_edit_distance_pairs, 10 * 1000, 5, is_max=False
     )
     dataset_info = mk_dataset_info("apps_min_edit_distance_prefrence")