prettify scores

folding_studio_demo/correlate.py

@@ -121,6 +121,9 @@ def plot_correlation_ranking(corr_data: pd.DataFrame, correlation_type: str) ->
 def fake_predict_and_correlate(spr_data_with_scores: pd.DataFrame, score_cols: list[str], main_cols: list[str]) -> tuple[pd.DataFrame, go.Figure]:
     """Fake predict structures of all complexes and correlate the results."""
     
+    # Rename score columns using the mapping in SCORE_COLUMN_NAMES
+    spr_data_with_scores = spr_data_with_scores.rename(columns=SCORE_COLUMN_NAMES)
+    
     corr_data = compute_correlation_data(spr_data_with_scores, score_cols)
     corr_ranking_plot = plot_correlation_ranking(corr_data, "Spearman")