Update src/about.py

src/about.py

@@ -111,28 +111,19 @@ benchmark_specific_metrics = {
     'affinity': ['mse', 'mae', 'corr']
 }
 
-display_labels = [
-    'BLAST', 'HMMER', 'K-SEP', 'APAAC', 'PFAM', 'AAC', 'PROTVEC',
-    'GENE2VEC', 'LEARNED-VEC', 'MUT2VEC', 'TCGA-EMBEDDING', 'CPCPROT', 
-    'SEQVEC', 'ProtBERT-BFD', 'TAPE-BERT-PFAM', 'ESM-1b', 'ProtALBERT', 'ProtXLNet', 'UNIREP', 'ProtT5-XL',
-    'MSA-Transformer'
-]
-
-group_colors = [
-    'green', 'green', 'green', 'green', 'green', 'green', 'blue',
-    'blue', 'blue', 'blue', 'blue', 'blue', 'red', 'red',
-    'red', 'red', 'red', 'red', 'red', 'red', 'red'
-]
-
-# Create a color dictionary from the labels and colors
-color_dict = {method.upper(): color for method, color in zip(display_labels, group_colors)}
-
 group_color_dict = {
     'INTERPRO2GO': 'orange', 'UNIRULE2GO': 'orange', 'ENSEMBL-ORTHOLOGY': 'orange',
     'K-SEP': 'green', 'BERT-PFAM': 'red', 'UNIREP': 'red', 'T5': 'red', 
     'BERT-BFD': 'red', 'HMMER': 'green', 'SEQVEC': 'red', 'ALBERT': 'red', 
-    'BLAST': 'green', 'PFAM': 'green', 'ESMB1': 'red', 'XLNET': 'red', 
-    'AAC': 'green', 'APAAC': 'green', 'PROTVEC': 'blue', 'MUT2VEC': 'blue',
+    'BLAST': 'green', 'PFAM': 'green', 'ESM1B': 'red', 'XLNET': 'red', 
+    'APAAC': 'green', 'APAAC': 'green', 'PROTVEC': 'blue', 'MUT2VEC': 'blue',
     'LEARNED-VEC': 'blue', 'CPC-PROT': 'blue', 'GENE2VEC': 'blue', 
     'TCGA-EMBEDDING': 'blue'
 }
+
+base_methods = [
+    'INTERPRO2GO', 'UNIRULE2GO', 'ENSEMBL-ORTHOLOGY', 'K-SEP', 'BERT-PFAM', 
+    'UNIREP', 'T5', 'BERT-BFD', 'HMMER', 'SEQVEC', 'ALBERT', 'BLAST', 'PFAM', 
+    'ESM1B', 'XLNET', 'APAAC', 'PROTVEC', 'MUT2VEC', 'LEARNED-VEC', 'CPC-PROT', 
+    'GENE2VEC', 'TCGA-EMBEDDING'
+]