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:
|