raised exceptions where they were not properly raised

MinuraPunchihewa · Jul 17, 2024 · 57485e2 · 57485e2
1 parent 326e40f
commit 57485e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hugging_py_face/nlp.py b/hugging_py_face/nlp.py
@@ -408,7 +408,7 @@ def translation(self, text: Union[Text, List], lang_input: Text = None, lang_out
         """
         if model is None:
             if lang_input is None or lang_output is None:
-                InsufficientParametersException("lang_input and lang_output are required if model is not provided.")
+                raise InsufficientParametersException("lang_input and lang_output are required if model is not provided.")
             model = f"{self.config['TASK_MODEL_MAP']['translation']}{lang_input}-{lang_output}"
             return self._query(text, options=options, model=model, task='translation')
         else:
@@ -428,7 +428,7 @@ def translation_in_df(self, df: DataFrame, column: Text, lang_input: Text = None
         """
         if model is None:
             if lang_input is None or lang_output is None:
-                InsufficientParametersException("lang_input and lang_output are required if model is not provided.")
+                raise InsufficientParametersException("lang_input and lang_output are required if model is not provided.")
             model = f"{self.config['TASK_MODEL_MAP']['translation']}{lang_input}-{lang_output}"
             predictions = self._query_in_df(df, column, options=options, model=model, task='translation')
         else: