From 88450a67989fd6a7f3493c73204fc204d309a9cc Mon Sep 17 00:00:00 2001 From: Santi Villalba Date: Sat, 16 Jul 2022 19:18:58 +0200 Subject: [PATCH] fix tag-sequence missmatch when predicting for multiple fastas --- run_pretrained_openfold.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/run_pretrained_openfold.py b/run_pretrained_openfold.py index 4c1f9753..c2384709 100644 --- a/run_pretrained_openfold.py +++ b/run_pretrained_openfold.py @@ -379,7 +379,7 @@ def main(args): # assert len(tags) == len(set(tags)), "All FASTA tags must be unique" tag = '-'.join(tags) - tag_list.append(tag) + tag_list.append((tag, tags)) seq_list.append(seqs) seq_sort_fn = lambda target: sum([len(s) for s in target[1]]) @@ -387,7 +387,7 @@ def main(args): feature_dicts = {} for model, output_directory in load_models_from_command_line(args, config): cur_tracing_interval = 0 - for tag, seqs in sorted_targets: + for (tag, tags), seqs in sorted_targets: output_name = f'{tag}_{args.config_preset}' if args.output_postfix is not None: output_name = f'{output_name}_{args.output_postfix}'