diff --git a/tests/models/resnet/test_modeling_tf_resnet.py b/tests/models/resnet/test_modeling_tf_resnet.py index 5f4eead8661cbf..1056ebc8eeac4f 100644 --- a/tests/models/resnet/test_modeling_tf_resnet.py +++ b/tests/models/resnet/test_modeling_tf_resnet.py @@ -41,7 +41,7 @@ from transformers import AutoFeatureExtractor -class ResNetModelTester: +class TFResNetModelTester: def __init__( self, parent, @@ -116,7 +116,7 @@ def prepare_config_and_inputs_for_common(self): @require_tf -class ResNetModelTest(TFModelTesterMixin, unittest.TestCase): +class TFResNetModelTest(TFModelTesterMixin, unittest.TestCase): """ Here we also overwrite some of the tests of test_modeling_common.py, as ResNet does not use input_ids, inputs_embeds, attention_mask and seq_length. @@ -131,7 +131,7 @@ class ResNetModelTest(TFModelTesterMixin, unittest.TestCase): has_attentions = False def setUp(self): - self.model_tester = ResNetModelTester(self) + self.model_tester = TFResNetModelTester(self) self.config_tester = ConfigTester(self, config_class=ResNetConfig, has_text_modality=False) def test_config(self): @@ -223,7 +223,7 @@ def prepare_img(): @require_tf @require_vision -class ResNetModelIntegrationTest(unittest.TestCase): +class TFResNetModelIntegrationTest(unittest.TestCase): @cached_property def default_feature_extractor(self): return (