Fix/remove column only when it exists (#105)

open-spaced-repetition · Apr 18, 2024 · 6aa6fd0 · 6aa6fd0
1 parent 11c4b3f
commit 6aa6fd0
Show file tree

Hide file tree

Showing 2 changed files with 2 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.6"
+version = "4.27.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
@@ -598,8 +598,7 @@ def create_time_series(
                 1,
             )
 
-            df["review_duration"] = df["review_duration"].astype(int)
-            df["review_state"] = df["review_state"].astype(int)
+            df.drop(columns=["review_duration", "review_state"], inplace=True)
 
         df["review_date"] = pd.to_datetime(df["review_time"] // 1000, unit="s")
         df["review_date"] = (
@@ -690,8 +689,6 @@ def cum_concat(x):
             columns=[
                 "review_time",
                 "card_id",
-                "review_duration",
-                "review_state",
                 "review_date",
                 "real_days",
                 "review_rating",