From 5fd3a84b97254e74918a6c14c2dab787f2775236 Mon Sep 17 00:00:00 2001 From: Milot Mirdita Date: Thu, 7 Dec 2023 18:22:30 +0900 Subject: [PATCH] Fix citations not correctly including alphafold --- colabfold/citations.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/colabfold/citations.py b/colabfold/citations.py index 6d01151d..7610b175 100644 --- 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"]