Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
609e6622
Unverified
Commit
609e6622
authored
Jun 19, 2024
by
amyeroberts
Committed by
GitHub
Jun 19, 2024
Browse files
Use self.config_tester.run_common_tests() (#31431)
* First testing updating config tests * Use run_common_tests
parent
7c71b61d
Changes
28
Hide whitespace changes
Inline
Side-by-side
Showing
8 changed files
with
58 additions
and
59 deletions
+58
-59
tests/models/superpoint/test_modeling_superpoint.py
tests/models/superpoint/test_modeling_superpoint.py
+8
-7
tests/models/swin/test_modeling_swin.py
tests/models/swin/test_modeling_swin.py
+8
-7
tests/models/swin2sr/test_modeling_swin2sr.py
tests/models/swin2sr/test_modeling_swin2sr.py
+8
-7
tests/models/swinv2/test_modeling_swinv2.py
tests/models/swinv2/test_modeling_swinv2.py
+8
-7
tests/models/timm_backbone/test_modeling_timm_backbone.py
tests/models/timm_backbone/test_modeling_timm_backbone.py
+6
-9
tests/models/univnet/test_modeling_univnet.py
tests/models/univnet/test_modeling_univnet.py
+4
-8
tests/models/upernet/test_modeling_upernet.py
tests/models/upernet/test_modeling_upernet.py
+8
-7
tests/models/vitmatte/test_modeling_vitmatte.py
tests/models/vitmatte/test_modeling_vitmatte.py
+8
-7
No files found.
tests/models/superpoint/test_modeling_superpoint.py
View file @
609e6622
...
...
@@ -124,15 +124,16 @@ class SuperPointModelTest(ModelTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
SuperPointModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
SuperPointConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
SuperPointConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"encoder_hidden_sizes"
,
"decoder_hidden_size"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"SuperPointForKeypointDetection does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
tests/models/swin/test_modeling_swin.py
View file @
609e6622
...
...
@@ -243,15 +243,16 @@ class SwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
SwinModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
SwinConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
SwinConfig
,
embed_dim
=
37
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/swin2sr/test_modeling_swin2sr.py
View file @
609e6622
...
...
@@ -175,15 +175,16 @@ class Swin2SRModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase)
def
setUp
(
self
):
self
.
model_tester
=
Swin2SRModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
Swin2SRConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
Swin2SRConfig
,
embed_dim
=
37
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/swinv2/test_modeling_swinv2.py
View file @
609e6622
...
...
@@ -229,15 +229,16 @@ class Swinv2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
Swinv2ModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
Swinv2Config
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
Swinv2Config
,
embed_dim
=
37
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/timm_backbone/test_modeling_timm_backbone.py
View file @
609e6622
...
...
@@ -18,7 +18,6 @@ import inspect
import
unittest
from
transformers
import
AutoBackbone
from
transformers.configuration_utils
import
PretrainedConfig
from
transformers.testing_utils
import
require_timm
,
require_torch
,
torch_device
from
transformers.utils.import_utils
import
is_torch_available
...
...
@@ -106,17 +105,15 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste
has_attentions
=
False
def
setUp
(
self
):
self
.
config_class
=
PretrainedConfig
# self.config_class = PretrainedConfig
self
.
config_class
=
TimmBackboneConfig
self
.
model_tester
=
TimmBackboneModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
self
.
config_class
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
self
.
config_class
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
]
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_timm_transformer_backbone_equivalence
(
self
):
timm_checkpoint
=
"resnet18"
...
...
tests/models/univnet/test_modeling_univnet.py
View file @
609e6622
...
...
@@ -121,20 +121,16 @@ class UnivNetModelTest(ModelTesterMixin, unittest.TestCase):
def
setUp
(
self
):
self
.
model_tester
=
UnivNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
UnivNetConfig
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
UnivNetConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_mel_bins"
]
)
@
unittest
.
skip
(
reason
=
"fix this once it gets more usage"
)
def
test_multi_gpu_data_parallel_forward
(
self
):
super
().
test_multi_gpu_data_parallel_forward
()
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained_subfolder
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/upernet/test_modeling_upernet.py
View file @
609e6622
...
...
@@ -161,15 +161,16 @@ class UperNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase)
def
setUp
(
self
):
self
.
model_tester
=
UperNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
UperNetConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
UperNetConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"hidden_size"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
def
test_for_semantic_segmentation
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/vitmatte/test_modeling_vitmatte.py
View file @
609e6622
...
...
@@ -146,15 +146,16 @@ class VitMatteModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
def
setUp
(
self
):
self
.
model_tester
=
VitMatteModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
VitMatteConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
VitMatteConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"hidden_size"
],
)
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
self
.
config_tester
.
run_common_tests
()
@
unittest
.
skip
(
reason
=
"VitMatte does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment