Fix/don't return empty dataframe in outlier filter (#95)

open-spaced-repetition · Mar 25, 2024 · 07ae177 · 07ae177
1 parent 02842aa
commit 07ae177
Show file tree

Hide file tree

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

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -427,8 +427,6 @@ def remove_outliers(group: pd.DataFrame) -> pd.DataFrame:
     ).index
 
     total = sum(grouped_group[("y", "count")])
-    if total <= 20:
-        return pd.DataFrame()
     has_been_removed = 0
     for i in sort_index:
         count = grouped_group.loc[i, ("y", "count")]