From 4e9a6796c7f2b26f5225f4ed3e8f74071947c2d0 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 21 Jun 2021 16:37:13 +0100 Subject: [PATCH] [Flax] Fix flax test save pretrained (#12256) * fix_torch_device_generate_test * remove @ * fix flax save pretrained test --- tests/test_modeling_flax_common.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/test_modeling_flax_common.py b/tests/test_modeling_flax_common.py index f2d30eea410..3f4e9edb5d0 100644 --- a/tests/test_modeling_flax_common.py +++ b/tests/test_modeling_flax_common.py @@ -247,9 +247,6 @@ class FlaxModelTesterMixin: config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() for model_class in self.all_model_classes: - if model_class.__name__ != "FlaxBertModel": - continue - with self.subTest(model_class.__name__): model = model_class(config)