correct split of paired and unpaired msas for trimer or higher

sokrypton · Feb 24, 2024 · f1f6a57 · f1f6a57
1 parent 2ecbcb3
commit f1f6a57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/colabfold/batch.py b/colabfold/batch.py
@@ -1117,11 +1117,11 @@ def unserialize_msa(
                 curr_seq_len += 1
             seqs_line.append(paired_seq)
 
-        # is sequence is paired add them to output
+        # if sequence is paired add them to output
         if (
             not is_single_protein
             and not is_homooligomer
-            and sum(has_amino_acid) == len(query_seq_len)
+            and sum(has_amino_acid) > 1 # at least 2 sequences are paired
         ):
             header_no_faster = header.replace(">", "")
             header_no_faster_split = header_no_faster.split("\t")