Fix citations not correctly including alphafold

sokrypton · Dec 28, 2023 · 5fd3a84 · 5fd3a84
1 parent 2515b3b
commit 5fd3a84
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/colabfold/citations.py b/colabfold/citations.py
@@ -129,11 +129,11 @@ def write_bibtex(
     bibtex_file: str = "cite.bibtex",
 ) -> Path:
     to_cite = ["Mirdita2021"]
-    if model == "AlphaFold2-ptm":
+    if model == "alphafold2_ptm" or model == "alphafold2":
         to_cite += ["Jumper2021"]
-    if model == "DeepFold-v1":
+    if model == "deepfold_v1":
         to_cite += ["Lee2023"]
-    if model.startswith("AlphaFold2-multimer"):
+    if model.startswith("alphafold2_multimer"):
         to_cite += ["Evans2021"]
     if use_msa:
         to_cite += ["Mirdita2019"]