add missing positional arg (#450)
Browse files
src/axolotl/utils/trainer.py
CHANGED
|
@@ -184,7 +184,7 @@ class AxolotlTrainer(Trainer):
|
|
| 184 |
rank=self.args.process_index,
|
| 185 |
batch_size=self.args.per_device_eval_batch_size,
|
| 186 |
)
|
| 187 |
-
return super()._get_eval_sampler()
|
| 188 |
|
| 189 |
def get_train_dataloader(self) -> Union[DataLoader, MultipackDistributedDataloader]:
|
| 190 |
if self.args.sample_packing:
|
|
|
|
| 184 |
rank=self.args.process_index,
|
| 185 |
batch_size=self.args.per_device_eval_batch_size,
|
| 186 |
)
|
| 187 |
+
return super()._get_eval_sampler(eval_dataset)
|
| 188 |
|
| 189 |
def get_train_dataloader(self) -> Union[DataLoader, MultipackDistributedDataloader]:
|
| 190 |
if self.args.sample_packing:
|