From eb597475c53d76e97d130d4f652938ef674a36ad Mon Sep 17 00:00:00 2001
From: Jarrett Ye <jarrett.ye@outlook.com>
Date: Tue, 5 Mar 2024 15:54:34 +0800
Subject: [PATCH] Fix/set pretrain_only by fsrs_items (#3051)

* Fix/set pretrain_only and insufficient reviews based on fsrs_items

* use review_count for FsrsInsufficientReviews

* Update weights.rs

* Update weights.rs
---
 rslib/src/scheduler/fsrs/weights.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rslib/src/scheduler/fsrs/weights.rs b/rslib/src/scheduler/fsrs/weights.rs
index daaea2cf8..adfa9cd4e 100644
--- a/rslib/src/scheduler/fsrs/weights.rs
+++ b/rslib/src/scheduler/fsrs/weights.rs
@@ -94,7 +94,7 @@ impl Collection {
         });
         let fsrs = FSRS::new(Some(current_weights))?;
         let mut weights =
-            fsrs.compute_parameters(items.clone(), revlogs.len() < 1000, Some(progress2))?;
+            fsrs.compute_parameters(items.clone(), fsrs_items < 1000, Some(progress2))?;
         let metrics = fsrs.universal_metrics(items, &weights, |_| true)?;
         if metrics.0 < metrics.1 {
             weights = current_weights.to_vec();