From 776e25c5ce1abfff9d38f42a2aa542e117079a66 Mon Sep 17 00:00:00 2001 From: Matt Date: Tue, 17 Sep 2024 17:01:01 +0100 Subject: [PATCH] Cleanup return_text and return_full_text options in TextGenerationPipeline --- tests/pipelines/test_pipelines_text_generation.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/pipelines/test_pipelines_text_generation.py b/tests/pipelines/test_pipelines_text_generation.py index d0091449e18cba..455e287490a4ba 100644 --- a/tests/pipelines/test_pipelines_text_generation.py +++ b/tests/pipelines/test_pipelines_text_generation.py @@ -435,8 +435,6 @@ def run_pipeline_test(self, text_generator, _): ], ) - with self.assertRaises(ValueError): - outputs = text_generator("test", return_full_text=True, return_text=True) with self.assertRaises(ValueError): outputs = text_generator("test", return_full_text=True, return_tensors=True) with self.assertRaises(ValueError):