From d906679c22c87ccedae5d7642771f257b316a3a5 Mon Sep 17 00:00:00 2001
From: Melih Yilmaz <32707537+melihyilmaz@users.noreply.github.com>
Date: Fri, 22 Sep 2023 16:30:03 -0700
Subject: [PATCH] Fix missing return value

---
 casanovo/casanovo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/casanovo/casanovo.py b/casanovo/casanovo.py
index d217ba24..0a1c3618 100644
--- a/casanovo/casanovo.py
+++ b/casanovo/casanovo.py
@@ -207,7 +207,7 @@ def train(
     provided by MassIVE-KB, from which to train a new Casnovo model.
     """
     output = setup_logging(output, verbosity)
-    config = setup_model(model, config, output, True)
+    config, model = setup_model(model, config, output, True)
     with ModelRunner(config, model) as runner:
         logger.info("Training a model from:")
         for peak_file in train_peak_path: