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

Remove pytest dependency. (#2093)

Browse Source
This commit is contained in:
Aymeric Augustin
2019-12-07 13:46:14 +01:00
committed by Julien Chaumond
parent 2670b0d682
commit 0cb163865a
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
transformers/tests/optimization_tf_test.py
View File

@@ -3,18 +3,19 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import unittest import unittest
import pytest
from transformers import is_tf_available from transformers import is_tf_available
from .utils import require_tf
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from tensorflow.python.eager import context from tensorflow.python.eager import context
from tensorflow.python.framework import ops from tensorflow.python.framework import ops
from transformers import (create_optimizer, GradientAccumulator) from transformers import (create_optimizer, GradientAccumulator)
else:
pytestmark = pytest.mark.skip("Require TensorFlow")
@require_tf
class OptimizationFTest(unittest.TestCase): class OptimizationFTest(unittest.TestCase):
def assertListAlmostEqual(self, list1, list2, tol): def assertListAlmostEqual(self, list1, list2, tol):
self.assertEqual(len(list1), len(list2)) self.assertEqual(len(list1), len(list2))