Skip to content

Add TF prefix to TF-Res test class #18481

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Aug 5, 2022
Merged

Add TF prefix to TF-Res test class #18481

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
18fa2d7
Add TF prefix to TF-Res test class
ydshieh Aug 5, 2022
71490a1
fix
ydshieh Aug 5, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions tests/models/resnet/test_modeling_tf_resnet.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
from transformers import AutoFeatureExtractor


class ResNetModelTester:
class TFResNetModelTester:
def __init__(
self,
parent,
Expand Down Expand Up @@ -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.
Expand All @@ -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):
Expand Down Expand Up @@ -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 (
Expand Down