From 6aa6fd0a1801c5ac3b2555d3260efee7534f7484 Mon Sep 17 00:00:00 2001
From: Jarrett Ye <jarrett.ye@outlook.com>
Date: Thu, 18 Apr 2024 23:43:02 +0800
Subject: [PATCH] Fix/remove column only when it exists (#105)

---
 pyproject.toml                       | 2 +-
 src/fsrs_optimizer/fsrs_optimizer.py | 5 +----
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index a577da7..7954678 100644
--- 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
index cd77863..4faf65a 100644
--- 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",