Skip to content

GitLab

Unverified Commit ad5045e3 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

add missing `require_tf` for `TFOPTGenerationTest` (#19010) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent d14af22c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/models/opt/test_modeling_tf_opt.py
View file @ ad5045e3
...@@ -315,6 +315,7 @@ class TFOPTEmbeddingsTest(unittest.TestCase): ...@@ -315,6 +315,7 @@ class TFOPTEmbeddingsTest(unittest.TestCase):
self.assertTrue(np.allclose(logits, logits_meta, atol=1e-4)) self.assertTrue(np.allclose(logits, logits_meta, atol=1e-4))
@require_tf
@slow @slow
class TFOPTGenerationTest(unittest.TestCase): class TFOPTGenerationTest(unittest.TestCase):
@property @property
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment