Skip to content

Commit 32b8b7a

Browse files
committed
fix
1 parent f1fc717 commit 32b8b7a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

machine/jobs/smt_engine_build_job.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,14 +85,14 @@ def _batch_inference(
8585
src_pretranslations = stack.enter_context(self._translation_file_service.get_source_pretranslations())
8686
writer = stack.enter_context(self._translation_file_service.open_target_pretranslation_writer())
8787
current_inference_step = 0
88-
phase_progress(ProgressStatus.from_step(current_inference_step, inference_step_count))
88+
phase_progress(ProgressStatus.from_step(current_inference_step, inference_step_count, "inference"))
8989
batch_size = self._config["inference_batch_size"]
9090
for pi_batch in batch(src_pretranslations, batch_size):
9191
if check_canceled is not None:
9292
check_canceled()
9393
_translate_batch(engine, pi_batch, writer)
9494
current_inference_step += len(pi_batch)
95-
phase_progress(ProgressStatus.from_step(current_inference_step, inference_step_count))
95+
phase_progress(ProgressStatus.from_step(current_inference_step, inference_step_count, "inference"))
9696

9797
def _save_model(self) -> None:
9898
logger.info("Saving model")

0 commit comments

Comments
 (0)