From 2ecbcb352b90faba181771888e506d19b8d7256b Mon Sep 17 00:00:00 2001 From: YoshitakaMo Date: Thu, 22 Feb 2024 18:28:24 +0900 Subject: [PATCH] fix renaming issue for multiple seqs in fasta format (#568) --- colabfold/mmseqs/search.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/colabfold/mmseqs/search.py b/colabfold/mmseqs/search.py index 024161f2..ea2a6bd6 100644 --- a/colabfold/mmseqs/search.py +++ b/colabfold/mmseqs/search.py @@ -419,12 +419,15 @@ def main(): unpaired_msa, paired_msa, query_sequences, query_seqs_cardinality ) args.base.joinpath(f"{job_number}.a3m").write_text(msa) - # add raw_jobname to the output file - os.rename( - args.base.joinpath(f"{job_number}.a3m"), - args.base.joinpath(f"{safe_filename(raw_jobname)}.a3m"), - ) + # rename a3m files + for job_number, (raw_jobname, query_sequences, query_seqs_cardinality) in enumerate(queries_unique): + os.rename( + args.base.joinpath(f"{job_number}.a3m"), + args.base.joinpath(f"{safe_filename(raw_jobname)}.a3m"), + ) + + # rename m8 files if args.use_templates: id = 0 for raw_jobname, query_sequences, query_seqs_cardinality in queries_unique: