From 223e4bfba3aaef902cc96810ba8947b04e7b4b28 Mon Sep 17 00:00:00 2001
From: Milot Mirdita <milot@mirdita.de>
Date: Tue, 14 Nov 2023 09:48:41 +0900
Subject: [PATCH] Fix issue with --num-models being None

---
 colabfold/batch.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/colabfold/batch.py b/colabfold/batch.py
index 4d9add46..98ec40dc 100644
--- a/colabfold/batch.py
+++ b/colabfold/batch.py
@@ -1680,9 +1680,7 @@ def main():
     msa_group = parser.add_argument_group("MSA arguments", "")
     msa_group.add_argument(
         "--msa-only",
-        action="store_const",
-        const=0,
-        dest="num_models",
+        action="store_true",
         help="Query and store MSAs from the MSA server without structure prediction",
     )
     msa_group.add_argument(
@@ -2000,6 +1998,9 @@ def main():
     queries, is_complex = get_queries(args.input, args.sort_queries_by)
     model_type = set_model_type(is_complex, args.model_type)
 
+    if args.msa_only:
+        args.num_models = 0
+
     if args.num_models > 0:
         download_alphafold_params(model_type, data_dir)