Fixing TranslationPipeline Hierarchy + return_name

2020-12-28 16:38:55 +01:00
parent c145a9e87b
commit f9a057392f
1 changed files with 2 additions and 2 deletions
@@ -202,7 +202,7 @@ class SummarizationPipeline(Text2TextGenerationPipeline):


@add_end_docstrings(PIPELINE_INIT_ARGS)
-class TranslationPipeline(Pipeline):
+class TranslationPipeline(Text2TextGenerationPipeline):
    """
    Translates from one language to another.

@@ -218,7 +218,7 @@ class TranslationPipeline(Pipeline):
        en_fr_translator("How old are you?")
    """

-    return_name = "translated"
+    return_name = "translation"

    def check_inputs(self, input_length: int, min_length: int, max_length: int):
        if input_length > 0.9 * max_length: