fix unpackdb process in use_templates

sokrypton · Aug 16, 2024 · 61ef8eb · 61ef8eb
1 parent 1ccca5a
commit 61ef8eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/colabfold/mmseqs/search.py b/colabfold/mmseqs/search.py
@@ -185,8 +185,9 @@ def mmseqs_search_monomer(
         run_mmseqs(mmseqs, ["rmdb", base.joinpath("final.a3m")])
 
         if use_templates:
-            run_mmseqs(mmseqs, ["unpackdb", base.joinpath("res_pdb.m8"), base.joinpath("."), "--unpack-name-mode", "0", "--unpack-suffix", ".m8"])
-            run_mmseqs(mmseqs, ["rmdb", base.joinpath("res_pdb.m8")])
+            run_mmseqs(mmseqs, ["unpackdb", base.joinpath("res_pdb"), base.joinpath("."), "--unpack-name-mode", "0", "--unpack-suffix", ".m8"])
+            if base.joinpath("res_pdb").exists():
+                run_mmseqs(mmseqs, ["rmdb", base.joinpath("res_pdb")])
 
     run_mmseqs(mmseqs, ["rmdb", base.joinpath("prof_res")])
     run_mmseqs(mmseqs, ["rmdb", base.joinpath("prof_res_h")])