[pytes collection] Fix flax test collection (#34004)

bit weird but to filter I had to use this

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)