Update src/bin/PROBE.py

src/bin/PROBE.py

@@ -17,6 +17,13 @@ def load_representation(multi_col_representation_vector_file_path):
 
 def run_probe(benchmarks, representation_name, representation_file_human, representation_file_affinity, similarity_tasks=["Sparse","200","500"], function_prediction_aspec="All_Aspects", function_prediction_dataset="All_Data_Sets", family_prediction_dataset=["nc","uc50","uc30","mm15"], detailed_output=False):
     print("\n\nPROBE (Protein RepresentatiOn Benchmark) run is started...\n\n")
+    print(type(benchmarks))
+    print(benchmarks)
+    if any(item in ['similarity', 'function', 'family', 'all'] for item in benchmarks):
+        print("AAAAAAAAA")
+    else:
+        print("BBBBBBBB")
+    
 
     if any(item in ['similarity', 'function', 'family', 'all'] for item in benchmarks):
         print("\nRepresentation vectors are loading...\n")