From 58cf7e7fedc767a0aaa7dca9b42ae9c1c6c22989 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 21 Aug 2023 04:10:19 -0400 Subject: [PATCH] add missing positional arg (#450) --- src/axolotl/utils/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py index 5245011f6..a7abe38b6 100644 --- a/src/axolotl/utils/trainer.py +++ b/src/axolotl/utils/trainer.py @@ -184,7 +184,7 @@ class AxolotlTrainer(Trainer): rank=self.args.process_index, batch_size=self.args.per_device_eval_batch_size, ) - return super()._get_eval_sampler() + return super()._get_eval_sampler(eval_dataset) def get_train_dataloader(self) -> Union[DataLoader, MultipackDistributedDataloader]: if self.args.sample_packing: