riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 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
Show Stats 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
import unittest
import pytest
from transformers import is_tf_available
from .utils import require_tf
if is_tf_available():
import tensorflow as tf
from tensorflow.python.eager import context
from tensorflow.python.framework import ops
from transformers import (create_optimizer, GradientAccumulator)
else:
pytestmark = pytest.mark.skip("Require TensorFlow")
@require_tf
class OptimizationFTest(unittest.TestCase):
def assertListAlmostEqual(self, list1, list2, tol):
self.assertEqual(len(list1), len(list2))