Skip to content

Commit 05fe8ec

Browse files
committed
Adjust NMT engine test
1 parent 0e3a0d5 commit 05fe8ec

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/jobs/test_nmt_engine_build_job.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ def test_run(decoy: Decoy) -> None:
5050
]
5151
assert pretranslations[0]["pretranslation_toks"] == ["please", ",", "i", "have", "booked", "a", "room", "."]
5252
assert len(pretranslations[0]["alignment"]) > 0
53+
else:
54+
assert pretranslations[0]["source_toks"] == []
55+
assert pretranslations[0]["pretranslation_toks"] == []
56+
assert len(pretranslations[0]["alignment"]) == 0
5357
decoy.verify(env.translation_file_service.save_model(Path("model.tar.gz"), "models/save-model.tar.gz"), times=1)
5458

5559

@@ -175,7 +179,7 @@ def open_target_pretranslation_writer(env: _TestEnvironment) -> Iterator[DictToJ
175179
"trg_lang": "en",
176180
"save_model": "save-model",
177181
"inference_batch_size": 100,
178-
"align_pretranslations": is_eflomal_available(),
182+
"align_pretranslations": True,
179183
}
180184
),
181185
self.nmt_model_factory,

0 commit comments

Comments
 (0)