Update from reviewer comments

sillsdev · Nov 21, 2023 · cf60661 · cf60661
1 parent 7f7d8e9
commit cf60661
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/machine/translation/huggingface/hugging_face_nmt_engine.py b/machine/translation/huggingface/hugging_face_nmt_engine.py
@@ -47,20 +47,14 @@ def __init__(
                 and src_lang not in cast(Any, self._tokenizer).lang_code_to_id
                 and src_lang not in additional_special_tokens
             ):
-                raise ValueError(
-                    f"'{src_lang}' is not a valid language code. This error can happen when "
-                    + "there is no matching training data and the language code is not in the NLLB-200."
-                )
+                raise ValueError(f"The specified model does not support the language code '{src_lang}'")
 
             if (
                 tgt_lang is not None
                 and tgt_lang not in cast(Any, self._tokenizer).lang_code_to_id
                 and tgt_lang not in additional_special_tokens
             ):
-                raise ValueError(
-                    f"'{tgt_lang}' is not a valid language code. This error can happen when "
-                    + "there is no matching training data and the language code is not in the NLLB-200."
-                )
+                raise ValueError(f"The specified model does not support the language code '{src_lang}'")
 
         self._pipeline = _TranslationPipeline(
             model=model,