From 90e2da881ceff53b1220bda39bee39f78718ec29 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 3130d9f9..70551dcd 100644 --- a/llms/mlx_lm/tuner/datasets.py +++ b/llms/mlx_lm/tuner/datasets.py @@ -113,7 +113,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)