[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

@@ -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)