From 322b529fe892a4b4babf11e59d44e8fb5637b81c Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 13 Sep 2022 16:18:06 +0200 Subject: [PATCH] add missing @require_tf --- tests/models/opt/test_modeling_tf_opt.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/opt/test_modeling_tf_opt.py b/tests/models/opt/test_modeling_tf_opt.py index 61d6aad53fc169..4e9972e4aaa195 100644 --- a/tests/models/opt/test_modeling_tf_opt.py +++ b/tests/models/opt/test_modeling_tf_opt.py @@ -315,6 +315,7 @@ def test_logits(self): self.assertTrue(np.allclose(logits, logits_meta, atol=1e-4)) +@require_tf @slow class TFOPTGenerationTest(unittest.TestCase): @property