From 69a12f6fee865d31a28178ba0c09c0f717d09387 Mon Sep 17 00:00:00 2001 From: Jarrett Ye Date: Fri, 8 Sep 2023 18:22:44 +0800 Subject: [PATCH] Fix/sort revlog in create_time_series --- pyproject.toml | 2 +- src/fsrs_optimizer/fsrs_optimizer.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 21e1549..681f1cf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "FSRS-Optimizer" -version = "4.13.4" +version = "4.13.5" readme = "README.md" dependencies = [ "matplotlib>=3.7.0", diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py index ec44fb5..9d0d249 100644 --- a/src/fsrs_optimizer/fsrs_optimizer.py +++ b/src/fsrs_optimizer/fsrs_optimizer.py @@ -378,6 +378,7 @@ def flags2str(flags: List[int]) -> str: def create_time_series(self, timezone: str, revlog_start_date: str, next_day_starts_at: int): """Step 2""" df = pd.read_csv("./revlog.csv") + df.sort_values(by=['card_id', 'review_time'], inplace=True, ignore_index=True) df['review_state'] = df['review_state'].map(lambda x: x if x != New else Learning) self.state_sequence = np.array(df['review_state']) self.duration_sequence = np.array(df['review_duration'])