Skip to content

Commit e2ec566

Browse files
committed
Fix appending of results for every dataset
1 parent 273db3f commit e2ec566

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

scripts/ProteinGym_runs/run_performance_tests_proteingym_hybrid_dca_llm.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,15 +62,14 @@ def compute_performances(mut_data, mut_sep=':', start_i: int = 0, already_tested
6262
esm_base_model = esm_base_model.to(device)
6363
MAX_WT_SEQUENCE_LENGTH = 2000
6464
get_vram()
65-
hybrid_perfs = []
6665
plt.figure(figsize=(40, 12))
6766
numbers_of_datasets = [i + 1 for i in range(len(mut_data.keys()))]
68-
delta_times = []
6967
for i, (dset_key, dset_paths) in enumerate(mut_data.items()):
7068
if i >= start_i and i not in already_tested_is: # i > 3 and i <21: #i == 18 - 1:
7169
start_time = time.time()
7270
print(f'\n{i+1}/{len(mut_data.items())}\n'
7371
f'===============================================================')
72+
hybrid_perfs = []
7473
csv_path = dset_paths['CSV_path']
7574
msa_path = dset_paths['MSA_path']
7675
wt_seq = dset_paths['WT_sequence']
@@ -249,7 +248,6 @@ def compute_performances(mut_data, mut_sep=':', start_i: int = 0, already_tested
249248
gc.collect()
250249

251250
dt = time.time() - start_time
252-
delta_times.append(dt)
253251
dset_hybrid_perfs_i = ''
254252
for hp in hybrid_perfs:
255253
dset_hybrid_perfs_i += f'{hp},'

0 commit comments

Comments
 (0)