From ad5045e3e39442f86a6b08cd87c04338134dd55a Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 13 Sep 2022 18:10:11 +0200 Subject: [PATCH] add missing `require_tf` for `TFOPTGenerationTest` (#19010) Co-authored-by: ydshieh --- 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 61d6aad53f..4e9972e4aa 100644 --- a/tests/models/opt/test_modeling_tf_opt.py +++ b/tests/models/opt/test_modeling_tf_opt.py @@ -315,6 +315,7 @@ class TFOPTEmbeddingsTest(unittest.TestCase): self.assertTrue(np.allclose(logits, logits_meta, atol=1e-4)) +@require_tf @slow class TFOPTGenerationTest(unittest.TestCase): @property