Fix plotting bug

Author: niklases

scripts/ProteinGym_runs/results/dca_esm_and_hybrid_opt_results.csv

@@ -10,7 +10,7 @@ No.,Dataset,N_Variants,N_Max_Muts,Untrained_Performance_DCA,Untrained_Performanc
 9,AACC1_PSEAI_Dandage_2018,1801,1,0.3180712414525488,0.45793953382550573,0.36853292069676097,0.31903154410021367,0.47419937564784326,0.34631222767421677,0.32040004121707966,0.49314677543321084,0.39686704499767456,0.2952042076988178,0.5709163888032727,0.5266664565178812,1801,1701,1601,801,1552
 10,ACE2_HUMAN_Chan_2020,2223,1,0.24320754065919856,0.1855938942334426,0.2613054581997969,0.24853053607437606,0.3415010482086907,0.3072698479146479,0.24107663144652855,0.40477658833433,0.4492083634466871,0.26777869508203456,0.612337520250314,0.6152894367115027,2223,2123,2023,1223,6839
 11,ADRB2_HUMAN_Jones_2020,7800,1,0.5187856047925657,0.5310582600087359,0.5151672046363325,0.5183272590334468,0.5354854601514117,0.5488487428556871,0.5174727374036995,0.534084922400466,0.5521903129844146,0.5147111407689821,0.5669706982620246,0.5540211657698451,7800,7700,7600,6800,4785
-12,AICDA_HUMAN_Gajula_2014_3cycles,209,1,0.41950521618921593,0.4075489898558927,0.274172920796004,0.4423419549584822,0.48860049929876914,0.5028511154022686,nan,nan,nan,nan,209,109,nan,nan,157
+12,AICDA_HUMAN_Gajula_2014_3cycles,209,1,0.41950521618921593,0.4075489898558927,0.274172920796004,0.4423419549584822,0.48860049929876914,0.5028511154022686,nan,nan,nan,nan,nan,nan,209,109,nan,nan,157
 13,AMIE_PSEAE_Wrenbeck_2017,6227,1,0.4865220422275132,0.617273943143269,0.48233745058184774,0.48552958575667343,0.6403749050924776,0.505282581879187,0.48566014345393765,0.6479991220564996,0.5479414916922898,0.48491145680826714,0.7218370499161034,0.661875527400551,6227,6127,6027,5227,3794
 14,ARGR_ECOLI_Tsuboyama_2023_1AOY,1287,1,0.42900190781111663,0.3625682795745455,0.6029023718184582,0.42804881901444286,0.5122536761399821,0.7507500982023315,0.4378963939956423,0.7022162095363621,0.8047078283426998,0.4640321210133655,0.8376616916139521,0.8940703605337751,1287,1187,1087,287,813
 15,B2L11_HUMAN_Dutta_2010_binding-Mcl-1,170,1,-0.09215236134822696,0.37716061821801294,0.2737196279944554,-0.3058472436083095,0.7524626016971393,0.7476686204181611,nan,nan,nan,nan,nan,nan,170,70,nan,nan,122

scripts/ProteinGym_runs/run_performance_tests_proteingym_hybrid_dca_llm.py

@@ -211,7 +211,8 @@ def compute_performances(mut_data, mut_sep=':', start_i: int = 0, already_tested
                         print(f'Only {len(fitnesses)} in total, splitting the data in N_Train = {len(y_train)} '
                               f'and N_Test = {len(y_test)} results in N_Test <= 50 variants - '
                               f'not getting performance for N_Train = {len(y_train)}...')
-                        hybrid_perfs.append(np.nan)
+                        for k in [np.nan, np.nan, np.nan]:
+                            hybrid_perfs.append(k)
                         ns_y_test.append(np.nan)
                         continue
                     get_vram()
@@ -495,11 +496,11 @@ def plot_csv_data(csv, plot_name):
             already_tested_is = []
 
 
-    compute_performances(
-        mut_data=combined_mut_data, 
-        start_i=start_i, 
-        already_tested_is=already_tested_is
-    )
+    #compute_performances(
+    #    mut_data=combined_mut_data, 
+    #    start_i=start_i, 
+    #    already_tested_is=already_tested_is
+    #)
 
 
     with open(out_results_csv, 'r') as fh: