Fix/skip group if it is empty (#98)

open-spaced-repetition · Mar 25, 2024 · 40d8601 · 40d8601
1 parent 5fb95e1
commit 40d8601
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "FSRS-Optimizer"
-version = "4.27.0"
+version = "4.27.1"
 readme = "README.md"
 dependencies = [
     "matplotlib>=3.7.0",

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -814,10 +814,11 @@ def pretrain(self, dataset=None, verbose=True):
             group = self.S0_dataset_group[
                 self.S0_dataset_group["r_history"] == first_rating
             ]
-            if group.empty and verbose:
-                tqdm.write(
-                    f"Not enough data for first rating {first_rating}. Expected at least 1, got 0."
-                )
+            if group.empty:
+                if verbose:
+                    tqdm.write(
+                        f"Not enough data for first rating {first_rating}. Expected at least 1, got 0."
+                    )
                 continue
             delta_t = group["delta_t"]
             recall = (group["y"]["mean"] * group["y"]["count"] + average_recall * 1) / (