From 0d312e6bdbd53e65c478b8ab07fb8520d7b9773d Mon Sep 17 00:00:00 2001
From: Melih Yilmaz <32707537+melihyilmaz@users.noreply.github.com>
Date: Fri, 22 Sep 2023 10:43:20 -0700
Subject: [PATCH] Use auto-downloaded weights

---
 casanovo/casanovo.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/casanovo/casanovo.py b/casanovo/casanovo.py
index 992cf566..d217ba24 100644
--- a/casanovo/casanovo.py
+++ b/casanovo/casanovo.py
@@ -133,7 +133,7 @@ def sequence(
     to sequence peptides.
     """
     output = setup_logging(output, verbosity)
-    config = setup_model(model, config, output, False)
+    config, model = setup_model(model, config, output, False)
     with ModelRunner(config, model) as runner:
         logger.info("Sequencing peptides from:")
         for peak_file in peak_path:
@@ -164,7 +164,7 @@ def evaluate(
     such as those provided by MassIVE-KB.
     """
     output = setup_logging(output, verbosity)
-    config = setup_model(model, config, output, False)
+    config, model = setup_model(model, config, output, False)
     with ModelRunner(config, model) as runner:
         logger.info("Sequencing and evaluating peptides from:")
         for peak_file in annotated_peak_path:
@@ -378,7 +378,7 @@ def setup_model(
     for key, value in config.items():
         logger.debug("%s = %s", str(key), str(value))
 
-    return config
+    return config, model
 
 
 def _get_model_weights() -> str: