Skip to content

Commit e385c5a

Browse files
ydshiehamyeroberts
authored andcommitted
Add TF prefix to TF-Res test class (huggingface#18481)
Co-authored-by: ydshieh <[email protected]>
1 parent 8cacf30 commit e385c5a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/models/resnet/test_modeling_tf_resnet.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
from transformers import AutoFeatureExtractor
4242

4343

44-
class ResNetModelTester:
44+
class TFResNetModelTester:
4545
def __init__(
4646
self,
4747
parent,
@@ -116,7 +116,7 @@ def prepare_config_and_inputs_for_common(self):
116116

117117

118118
@require_tf
119-
class ResNetModelTest(TFModelTesterMixin, unittest.TestCase):
119+
class TFResNetModelTest(TFModelTesterMixin, unittest.TestCase):
120120
"""
121121
Here we also overwrite some of the tests of test_modeling_common.py, as ResNet does not use input_ids, inputs_embeds,
122122
attention_mask and seq_length.
@@ -131,7 +131,7 @@ class ResNetModelTest(TFModelTesterMixin, unittest.TestCase):
131131
has_attentions = False
132132

133133
def setUp(self):
134-
self.model_tester = ResNetModelTester(self)
134+
self.model_tester = TFResNetModelTester(self)
135135
self.config_tester = ConfigTester(self, config_class=ResNetConfig, has_text_modality=False)
136136

137137
def test_config(self):
@@ -223,7 +223,7 @@ def prepare_img():
223223

224224
@require_tf
225225
@require_vision
226-
class ResNetModelIntegrationTest(unittest.TestCase):
226+
class TFResNetModelIntegrationTest(unittest.TestCase):
227227
@cached_property
228228
def default_feature_extractor(self):
229229
return (

0 commit comments

Comments
 (0)