From 69282ab7fc1665f3ddc731fc88f727ebb02f7c53 Mon Sep 17 00:00:00 2001 From: Chime Ogbuji Date: Wed, 6 Nov 2024 12:58:00 -0500 Subject: [PATCH] Minor fix --- llms/mlx_lm/tuner/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/mlx_lm/tuner/datasets.py b/llms/mlx_lm/tuner/datasets.py index f1345634..c8b39e31 100644 --- a/llms/mlx_lm/tuner/datasets.py +++ b/llms/mlx_lm/tuner/datasets.py @@ -116,7 +116,7 @@ class CompletionsDatasetCollection: def get_prompt_and_completion(self, idx: int): def getitem(dataset: CompletionsDataset, index: int): - dataset.get_prompt_and_completion(index) + return dataset.get_prompt_and_completion(index) return self.__fetch_and_process_item__(idx, getitem)