Fix/set default weights in rmse_matrix

open-spaced-repetition · Dec 25, 2024 · eac51f9 · eac51f9
1 parent 4970509
commit eac51f9
Show file tree

Hide file tree

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

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -2109,6 +2109,8 @@ def count_lapse(r_history, t_history):
             else 0
         )
     )
+    if "weights" not in tmp.columns:
+        tmp["weights"] = 1
     tmp = (
         tmp.groupby(["delta_t", "i", "lapse"])
         .agg({"y": "mean", "p": "mean", "weights": "sum"})