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

Skip test while the model is not available (#12739)

Browse Source
This commit is contained in:
Lysandre Debut
2021-07-15 15:06:47 +02:00
committed by GitHub
parent 3290315a2a
commit 8c7bd1b97b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_megatron_bert.py
View File

@@ -356,6 +356,7 @@ TOLERANCE = 1e-4
@require_tokenizers @require_tokenizers
class MegatronBertModelIntegrationTests(unittest.TestCase): class MegatronBertModelIntegrationTests(unittest.TestCase):
@slow @slow
@unittest.skip("Model is not available.")
def test_inference_no_head(self): def test_inference_no_head(self):
directory = "nvidia/megatron-bert-uncased-345m" directory = "nvidia/megatron-bert-uncased-345m"
if "MYDIR" in os.environ: if "MYDIR" in os.environ: