Fix missing return value

Noble-Lab · Sep 22, 2023 · d906679 · d906679
1 parent 4ab4728
commit d906679
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: