SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Skip failing test

Browse Source
This commit is contained in:
Sylvain Gugger
2022-11-22 09:53:56 -05:00
parent 624ae09f5c
commit f3a1efd1cf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/plbart/test_modeling_plbart.py
View File

@@ -409,7 +409,7 @@ class PLBartBaseIntegrationTest(AbstractSeq2SeqIntegrationTest):
src_text = ["Is 0 the first Fibonacci number ?", "Find the sum of all prime numbers ."] src_text = ["Is 0 the first Fibonacci number ?", "Find the sum of all prime numbers ."]
tgt_text = ["0 the first Fibonacci number?", "the sum of all prime numbers.......... the the"] tgt_text = ["0 the first Fibonacci number?", "the sum of all prime numbers.......... the the"]
# @unittest.skip("This test is broken, still generates english") @unittest.skip("This test is broken, fix me gante")
def test_base_generate(self): def test_base_generate(self):
inputs = self.tokenizer([self.src_text[0]], return_tensors="pt").to(torch_device) inputs = self.tokenizer([self.src_text[0]], return_tensors="pt").to(torch_device)
translated_tokens = self.model.generate( translated_tokens = self.model.generate(