Format-The-Codebase

- rename a test
This commit is contained in:
Ömer Faruk Özdemir 2022-02-09 16:15:01 +03:00
parent 80b5e117c9
commit 3a9d018eab

View File

@ -243,7 +243,7 @@ class TestLoadInterface(unittest.TestCase):
class TestLoadFromPipeline(unittest.TestCase):
def test_question_answering(self):
def test_text_to_text_model_from_pipeline(self):
pipe = transformers.pipeline(model="sshleifer/bart-tiny-random")
output = pipe("My name is Sylvain and I work at Hugging Face in Brooklyn")
self.assertIsNotNone(output)