add missing positional arg (#450)

2023-08-21 04:10:19 -04:00
parent 04a42b6db1
commit 58cf7e7fed
1 changed files with 1 additions and 1 deletions
--- 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: