Update src/vis_utils.py

src/vis_utils.py

@@ -38,18 +38,18 @@ def set_colors_and_marks_for_representation_groups(ax):
             label.set_text(f"^ {text}")
 
 def benchmark_plot(benchmark_type, methods_selected, x_metric, y_metric):
-    if benchmark_type == 'flexible':
-        # Use general visualizer logic
-        return general_visualizer_plot(methods_selected, x_metric=x_metric, y_metric=y_metric)
-    elif benchmark_type == 'similarity':
+    if benchmark_type == 'similarity':
         title = f"{x_metric} vs {y_metric}"
-        return draw_scatter_plot_similarity(methods_selected, x_metric, y_metric, title)
+        return plot_similarity_results(methods_selected, x_metric, y_metric, title)
     elif benchmark_type == 'function':
         return plot_function_results("./data/function_results.csv", x_metric, y_metric, methods_selected)
     elif benchmark_type == 'family':
         return plot_family_results("./data/family_results.csv", methods_selected, x_metric, save_path="./plot_images")
     elif benchmark_type == "affinity":
         return plot_affinity_results("./data/affinity_results.csv", methods_selected, x_metric, save_path="./plot_images")
+    else:
+        # Use general visualizer logic
+        return general_visualizer_plot(methods_selected, x_metric=x_metric, y_metric=y_metric)
 
 def general_visualizer(methods_selected, x_metric, y_metric):
     df = pd.read_csv(CSV_RESULT_PATH)