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

Require Torch when testing examples

Browse Source
This commit is contained in:
Lysandre
2020-01-14 16:31:39 -05:00
committed by Lysandre Debut
parent 9bab9b83d2
commit eaa6b9afc6
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_examples.py
View File

@@ -15,6 +15,7 @@
import os import os
import unittest import unittest
from .utils import require_torch
def get_examples_from_file(file): def get_examples_from_file(file):
@@ -40,6 +41,7 @@ def get_examples_from_file(file):
return ['\n'.join(example) for example in examples] return ['\n'.join(example) for example in examples]
@require_torch
class TestCodeExamples(unittest.TestCase): class TestCodeExamples(unittest.TestCase):
def test_configuration_examples(self): def test_configuration_examples(self):
transformers_directory = "../src/transformers" transformers_directory = "../src/transformers"