winglian commited on
Commit
58cf7e7
1 Parent(s): 04a42b6

add missing positional arg (#450)

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/trainer.py +1 -1
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: