Fix panic when training interrupted

ankitects/anki#2706
open-spaced-repetition · Oct 13, 2023 · 1d0ca1a · 1d0ca1a
1 parent 5089470
commit 1d0ca1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/training.rs b/src/training.rs
@@ -236,9 +236,9 @@ impl<B: Backend> FSRS<B> {
                     self.device(),
                     progress.clone().map(ProgressCollector::new),
                 );
-                model.unwrap().w.val().to_data().convert().value
+                Ok(model?.w.val().to_data().convert().value)
             })
-            .collect();
+            .collect::<Result<_>>()?;
 
         let average_weights = weights_sets
             .iter()