Update forward signature test for vision models (#27681)

* Update forward signature

* Empty-Commit
This commit is contained in:
NielsRogge
2023-11-27 15:48:17 +01:00
committed by GitHub
parent 1d7f406e19
commit 59499bbe8b
48 changed files with 1 additions and 612 deletions

View File

@@ -15,7 +15,6 @@
""" Testing suite for the PyTorch Donut Swin model. """
import collections
import inspect
import unittest
from transformers import DonutSwinConfig
@@ -186,18 +185,6 @@ class DonutSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
x = model.get_output_embeddings()
self.assertTrue(x is None or isinstance(x, nn.Linear))
def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()
for model_class in self.all_model_classes:
model = model_class(config)
signature = inspect.signature(model.forward)
# signature.parameters is an OrderedDict => so arg_names order is deterministic
arg_names = [*signature.parameters.keys()]
expected_arg_names = ["pixel_values"]
self.assertListEqual(arg_names[:1], expected_arg_names)
def test_attention_outputs(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
config.return_dict = True