Fix always is_custom_template.

ericmjl · Mar 1, 2024 · 10b6838 · 10b6838
1 parent 17f24bd
commit 10b6838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_pretrained_openfold.py b/run_pretrained_openfold.py
@@ -186,7 +186,7 @@ def main(args):
             )
 
     is_multimer = "multimer" in args.config_preset
-    is_custom_template = "use_custom_template" in args
+    is_custom_template = "use_custom_template" in args and args.use_custom_template
     if is_custom_template:
         template_featurizer = templates.CustomHitFeaturizer(
             mmcif_dir=args.template_mmcif_dir,