Logging instead of print

chaidiscovery · Nov 28, 2024 · 3c60568 · 3c60568
1 parent b07816e
commit 3c60568
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chai_lab/data/dataset/msas/colabfold.py b/chai_lab/data/dataset/msas/colabfold.py
@@ -366,7 +366,7 @@ def generate_colabfold_msas(protein_seqs: list[str], msa_dir: Path):
         a3ms_dir.mkdir()
 
         # Generate MSAs for each protein chain
-        print(f"Running MSA generation for {len(protein_seqs)} protein sequences")
+        logger.info(f"Running MSA generation for {len(protein_seqs)} protein sequences")
         msas = _run_mmseqs2(
             protein_seqs,
             mmseqs_dir,