From 74b846cacfc5cade3454f56ce43f7503763606da Mon Sep 17 00:00:00 2001
From: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Date: Tue, 13 Jun 2023 14:28:14 +0200
Subject: [PATCH] Update `(TF)SamModelIntegrationTest` (#24199)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
---
 tests/models/sam/test_modeling_tf_sam.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/models/sam/test_modeling_tf_sam.py b/tests/models/sam/test_modeling_tf_sam.py
index 33cb38c6fb..863d4bd842 100644
--- a/tests/models/sam/test_modeling_tf_sam.py
+++ b/tests/models/sam/test_modeling_tf_sam.py
@@ -428,8 +428,9 @@ def prepare_dog_img():
     return raw_image
 
 
+@require_tf
 @slow
-class SamModelIntegrationTest(unittest.TestCase):
+class TFSamModelIntegrationTest(unittest.TestCase):
     def tearDown(self):
         super().tearDown()
         # clean-up as much as possible GPU memory occupied by PyTorch