Fix template search (#565)

* Fix template search * Update search.py Allow the usage of the precomputed index for template search
sokrypton · Jan 31, 2024 · 0cf2621 · 0cf2621
1 parent a00ce1b
commit 0cf2621
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/colabfold/mmseqs/search.py b/colabfold/mmseqs/search.py
@@ -74,9 +74,11 @@ def mmseqs_search_monomer(
             db_load_mode = 0
             dbSuffix1 = "_seq"
             dbSuffix2 = "_aln"
+            dbSuffix3 = ""
         else:
             dbSuffix1 = ".idx"
             dbSuffix2 = ".idx"
+            dbSuffix3 = ".idx"
 
     # fmt: off
     # @formatter:off
@@ -108,7 +110,7 @@ def mmseqs_search_monomer(
     if use_templates:
         run_mmseqs(mmseqs, ["search", base.joinpath("prof_res"), dbbase.joinpath(template_db), base.joinpath("res_pdb"),
                             base.joinpath("tmp2"), "--db-load-mode", str(db_load_mode), "--threads", str(threads), "-s", "7.5", "-a", "-e", "0.1"])
-        run_mmseqs(mmseqs, ["convertalis", base.joinpath("prof_res"), dbbase.joinpath(f"{template_db}{dbSuffix1}"), base.joinpath("res_pdb"),
+        run_mmseqs(mmseqs, ["convertalis", base.joinpath("prof_res"), dbbase.joinpath(f"{template_db}{dbSuffix3}"), base.joinpath("res_pdb"),
                             base.joinpath(f"{template_db}.m8"), "--format-output",
                             "query,target,fident,alnlen,mismatch,gapopen,qstart,qend,tstart,tend,evalue,bits,cigar",
                             "--db-load-mode", str(db_load_mode), "--threads", str(threads)])