fix: report progress correctly for searchers configured in epochs
This fixes the calls to `report_progress` when the searcher is configured in terms of epochs to report absolute epochs, not relative percent complete w.r.t. total epochs requested.
@@ -223,7 +223,7 @@ class WorkloadSequencer(workload.Source):
elif op.unit == _generic.Unit.RECORDS:
op.report_progress(self.global_batch_size * self.state.latest_batch)
elif op.unit == _generic.Unit.EPOCHS:
- op.report_progress(self.state.latest_batch / self.as_batches(epochs=op.epochs))
+ op.report_progress(self.state.latest_batch / self.as_batches(epochs=1))
else:
raise ValueError(f"unrecognized searcher op unit: {op.unit}")
