Fix/don't sort by length in batch_predict() (#93)

open-spaced-repetition · Mar 21, 2024 · 08eb5d1 · 08eb5d1
1 parent 9da7847
commit 08eb5d1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 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.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
@@ -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[