Added "Copied from" comments in test_modeling_pvt_v2.py

Author: FoamoftheSea

tests/models/pvt_v2/test_modeling_pvt_v2.py

@@ -47,6 +47,7 @@
     from PIL import Image
 
 
+# Copied from tests.models.pvt.test_modeling_pvt
 class PvtV2ConfigTester(ConfigTester):
     def run_common_tests(self):
         config = self.config_class(**self.inputs_dict)
@@ -123,13 +124,15 @@ def get_config(self):
             out_indices=self.out_indices,
         )
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def create_and_check_model(self, config, pixel_values, labels):
         model = PvtV2Model(config=config)
         model.to(torch_device)
         model.eval()
         result = model(pixel_values)
         self.parent.assertIsNotNone(result.last_hidden_state)
 
+    # Copied from tests.models.resnet.test_modeling_resnet
     def create_and_check_backbone(self, config, pixel_values, labels):
         model = PvtV2Backbone(config=config)
         model.to(torch_device)
@@ -177,6 +180,7 @@ def create_and_check_for_image_classification(self, config, pixel_values, labels
         result = model(pixel_values)
         self.parent.assertEqual(result.logits.shape, (self.batch_size, self.type_sequence_label_size))
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def prepare_config_and_inputs_for_common(self):
         config_and_inputs = self.prepare_config_and_inputs()
         config, pixel_values, labels = config_and_inputs
@@ -185,6 +189,7 @@ def prepare_config_and_inputs_for_common(self):
 
 
 # We will verify our results on an image of cute cats
+# Copied from tests.models.pvt.test_modeling_pvt
 def prepare_img():
     image = Image.open("./tests/fixtures/tests_samples/COCO/000000039769.png")
     return image
@@ -205,21 +210,26 @@ class PvtV2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
     test_torchscript = False
     has_attentions = False
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def setUp(self):
         self.model_tester = PvtV2ModelTester(self)
         self.config_tester = PvtV2ConfigTester(self, config_class=PvtV2Config)
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def test_config(self):
         self.config_tester.run_common_tests()
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def test_model(self):
         config_and_inputs = self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_model(*config_and_inputs)
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     @unittest.skip("Pvt-V2 does not use inputs_embeds")
     def test_inputs_embeds(self):
         pass
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     @unittest.skip("Pvt-V2 does not have get_input_embeddings method and get_output_embeddings methods")
     def test_model_common_attributes(self):
         pass
@@ -235,6 +245,7 @@ def test_training_gradient_checkpointing_use_reentrant(self):
         # torch.utils.checkpoint.checkpoint
         self.check_training_gradient_checkpointing(gradient_checkpointing_kwargs={"use_reentrant": True})
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def test_initialization(self):
         config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
 
@@ -282,6 +293,7 @@ def check_hidden_states_output(inputs_dict, config, model_class):
 
             check_hidden_states_output(inputs_dict, config, model_class)
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     def test_training(self):
         if not self.model_tester.is_training:
             return
@@ -311,6 +323,7 @@ def test_forward_signature(self):
             expected_arg_names = ["pixel_values"]
             self.assertListEqual(arg_names[:1], expected_arg_names)
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     @slow
     def test_model_from_pretrained(self):
         for model_name in PVT_V2_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
@@ -363,6 +376,7 @@ def test_inference_model(self):
 
         self.assertTrue(torch.allclose(outputs.last_hidden_state[0, :3, :3], expected_slice, atol=1e-4))
 
+    # Copied from tests.models.pvt.test_modeling_pvt
     @slow
     @require_accelerate
     @require_torch_accelerator