From 736c7cde51df63a49aefe7dd333dac39ff107010 Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Mon, 7 Oct 2024 18:11:13 +0200 Subject: [PATCH] [`pytes collection`] Fix flax test collection (#34004) bit weird but to filter I had to use this --- tests/models/albert/test_modeling_flax_albert.py | 1 + tests/models/beit/test_modeling_flax_beit.py | 1 + tests/models/bert/test_modeling_flax_bert.py | 1 + tests/models/big_bird/test_modeling_flax_big_bird.py | 1 + tests/models/distilbert/test_modeling_flax_distilbert.py | 1 + tests/models/electra/test_modeling_flax_electra.py | 1 + tests/models/mbart/test_modeling_flax_mbart.py | 1 + tests/models/regnet/test_modeling_flax_regnet.py | 1 + tests/models/resnet/test_modeling_flax_resnet.py | 1 + tests/models/roberta/test_modeling_flax_roberta.py | 1 + .../test_modeling_flax_roberta_prelayernorm.py | 1 + tests/models/roformer/test_modeling_flax_roformer.py | 1 + tests/models/vit/test_modeling_flax_vit.py | 1 + 13 files changed, 13 insertions(+) diff --git a/tests/models/albert/test_modeling_flax_albert.py b/tests/models/albert/test_modeling_flax_albert.py index 956de9ebdc..90590e737f 100644 --- a/tests/models/albert/test_modeling_flax_albert.py +++ b/tests/models/albert/test_modeling_flax_albert.py @@ -80,6 +80,7 @@ class FlaxAlbertModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/beit/test_modeling_flax_beit.py b/tests/models/beit/test_modeling_flax_beit.py index 78c24220c2..24307532fd 100644 --- a/tests/models/beit/test_modeling_flax_beit.py +++ b/tests/models/beit/test_modeling_flax_beit.py @@ -79,6 +79,7 @@ class FlaxBeitModelTester(unittest.TestCase): # in BeiT, the seq length equals the number of patches + 1 (we add 1 for the [CLS] token) num_patches = (image_size // patch_size) ** 2 self.seq_length = num_patches + 1 + super().__init__() def prepare_config_and_inputs(self): pixel_values = floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size]) diff --git a/tests/models/bert/test_modeling_flax_bert.py b/tests/models/bert/test_modeling_flax_bert.py index fca54dbed8..4a9610d723 100644 --- a/tests/models/bert/test_modeling_flax_bert.py +++ b/tests/models/bert/test_modeling_flax_bert.py @@ -79,6 +79,7 @@ class FlaxBertModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/big_bird/test_modeling_flax_big_bird.py b/tests/models/big_bird/test_modeling_flax_big_bird.py index 63b2237fbd..f889952d2b 100644 --- a/tests/models/big_bird/test_modeling_flax_big_bird.py +++ b/tests/models/big_bird/test_modeling_flax_big_bird.py @@ -90,6 +90,7 @@ class FlaxBigBirdModelTester(unittest.TestCase): self.use_bias = use_bias self.block_size = block_size self.num_random_blocks = num_random_blocks + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/distilbert/test_modeling_flax_distilbert.py b/tests/models/distilbert/test_modeling_flax_distilbert.py index 1f5a402e86..39a25a42fe 100644 --- a/tests/models/distilbert/test_modeling_flax_distilbert.py +++ b/tests/models/distilbert/test_modeling_flax_distilbert.py @@ -79,6 +79,7 @@ class FlaxDistilBertModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/electra/test_modeling_flax_electra.py b/tests/models/electra/test_modeling_flax_electra.py index 19b35d8940..f531c7f8d0 100644 --- a/tests/models/electra/test_modeling_flax_electra.py +++ b/tests/models/electra/test_modeling_flax_electra.py @@ -67,6 +67,7 @@ class FlaxElectraModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/mbart/test_modeling_flax_mbart.py b/tests/models/mbart/test_modeling_flax_mbart.py index a642b2344c..ef48e7c4f3 100644 --- a/tests/models/mbart/test_modeling_flax_mbart.py +++ b/tests/models/mbart/test_modeling_flax_mbart.py @@ -116,6 +116,7 @@ class FlaxMBartModelTester(unittest.TestCase): self.bos_token_id = bos_token_id self.decoder_start_token_id = decoder_start_token_id self.initializer_range = initializer_range + super().__init__() def prepare_config_and_inputs(self): input_ids = np.clip(ids_tensor([self.batch_size, self.seq_length - 1], self.vocab_size), 3, self.vocab_size) diff --git a/tests/models/regnet/test_modeling_flax_regnet.py b/tests/models/regnet/test_modeling_flax_regnet.py index 911d595c56..314f0b367b 100644 --- a/tests/models/regnet/test_modeling_flax_regnet.py +++ b/tests/models/regnet/test_modeling_flax_regnet.py @@ -65,6 +65,7 @@ class FlaxRegNetModelTester(unittest.TestCase): self.num_labels = num_labels self.scope = scope self.num_stages = len(hidden_sizes) + super().__init__() def prepare_config_and_inputs(self): pixel_values = floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size]) diff --git a/tests/models/resnet/test_modeling_flax_resnet.py b/tests/models/resnet/test_modeling_flax_resnet.py index e9566e2e2f..ce83d415dc 100644 --- a/tests/models/resnet/test_modeling_flax_resnet.py +++ b/tests/models/resnet/test_modeling_flax_resnet.py @@ -64,6 +64,7 @@ class FlaxResNetModelTester(unittest.TestCase): self.num_labels = num_labels self.scope = scope self.num_stages = len(hidden_sizes) + super().__init__() def prepare_config_and_inputs(self): pixel_values = floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size]) diff --git a/tests/models/roberta/test_modeling_flax_roberta.py b/tests/models/roberta/test_modeling_flax_roberta.py index d205a0e75f..f2f7296df6 100644 --- a/tests/models/roberta/test_modeling_flax_roberta.py +++ b/tests/models/roberta/test_modeling_flax_roberta.py @@ -78,6 +78,7 @@ class FlaxRobertaModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/roberta_prelayernorm/test_modeling_flax_roberta_prelayernorm.py b/tests/models/roberta_prelayernorm/test_modeling_flax_roberta_prelayernorm.py index 0074323460..409752e162 100644 --- a/tests/models/roberta_prelayernorm/test_modeling_flax_roberta_prelayernorm.py +++ b/tests/models/roberta_prelayernorm/test_modeling_flax_roberta_prelayernorm.py @@ -81,6 +81,7 @@ class FlaxRobertaPreLayerNormModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/roformer/test_modeling_flax_roformer.py b/tests/models/roformer/test_modeling_flax_roformer.py index 8364e121b4..971c1a18cd 100644 --- a/tests/models/roformer/test_modeling_flax_roformer.py +++ b/tests/models/roformer/test_modeling_flax_roformer.py @@ -79,6 +79,7 @@ class FlaxRoFormerModelTester(unittest.TestCase): self.type_sequence_label_size = type_sequence_label_size self.initializer_range = initializer_range self.num_choices = num_choices + super().__init__() def prepare_config_and_inputs(self): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) diff --git a/tests/models/vit/test_modeling_flax_vit.py b/tests/models/vit/test_modeling_flax_vit.py index fb53caa343..97fc3082a9 100644 --- a/tests/models/vit/test_modeling_flax_vit.py +++ b/tests/models/vit/test_modeling_flax_vit.py @@ -72,6 +72,7 @@ class FlaxViTModelTester(unittest.TestCase): # in ViT, the seq length equals the number of patches + 1 (we add 1 for the [CLS] token) num_patches = (image_size // patch_size) ** 2 self.seq_length = num_patches + 1 + super().__init__() def prepare_config_and_inputs(self): pixel_values = floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size])