From 08eb5d185f43227ada25a7a2e28f898bb31c54cd Mon Sep 17 00:00:00 2001
From: Jarrett Ye <jarrett.ye@outlook.com>
Date: Thu, 21 Mar 2024 14:12:19 +0800
Subject: [PATCH] Fix/don't sort by length in batch_predict() (#93)

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

diff --git a/pyproject.toml b/pyproject.toml
index 55e688d..1f24e96 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "FSRS-Optimizer"
-version = "4.26.4"
+version = "4.26.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 617b5c7..31bf922 100644
--- a/src/fsrs_optimizer/fsrs_optimizer.py
+++ b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -400,7 +400,7 @@ def predict(self, t_history: str, r_history: str):
             return output_t[-1][0]
 
     def batch_predict(self, dataset):
-        fast_dataset = BatchDataset(dataset)
+        fast_dataset = BatchDataset(dataset, sort_by_length=False)
         with torch.no_grad():
             outputs, _ = self.model(fast_dataset.x_train.transpose(0, 1))
             stabilities, difficulties = outputs[