Skip to content

Commit

Permalink
Cleanup return_text and return_full_text options in TextGenerationPip…
Browse files Browse the repository at this point in the history
…eline
  • Loading branch information
Rocketknight1 committed Sep 26, 2024
1 parent a27dbaf commit 776e25c
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions tests/pipelines/test_pipelines_text_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 776e25c

Please sign in to comment.