Skip to content

Commit

Permalink
[CI-test_torch] skip test_tf_from_pt_safetensors for 4 models (#2…
Browse files Browse the repository at this point in the history
…7481)

* skip 4 tests

* nits

* style

* wow it's not my day
  • Loading branch information
ArthurZucker authored Nov 14, 2023
1 parent d71fa9f commit e107ae3
Show file tree
Hide file tree
Showing 4 changed files with 16 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/models/mobilebert/test_modeling_mobilebert.py
Original file line number Diff line number Diff line change
Expand Up @@ -302,6 +302,10 @@ def _prepare_for_class(self, inputs_dict, model_class, return_labels=False):
def test_resize_tokens_embeddings(self):
super().test_resize_tokens_embeddings()

@unittest.skip("This test is currently broken because of safetensors.")
def test_tf_from_pt_safetensors(self):
pass

def setUp(self):
self.model_tester = MobileBertModelTester(self)
self.config_tester = ConfigTester(self, config_class=MobileBertConfig, hidden_size=37)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -196,6 +196,10 @@ def setUp(
def test_inputs_embeds(self):
pass

@unittest.skip("This test is currently broken because of safetensors.")
def test_tf_from_pt_safetensors(self):
pass

# speech2text2 has no base model
def test_save_load_fast_init_from_base(self):
pass
Expand Down
4 changes: 4 additions & 0 deletions tests/models/transfo_xl/test_modeling_transfo_xl.py
Original file line number Diff line number Diff line change
Expand Up @@ -490,6 +490,10 @@ def _mock_init_weights(self, module):
def test_left_padding_compatibility(self):
pass

@unittest.skip("This test is currently broken because of safetensors.")
def test_tf_from_pt_safetensors(self):
pass


@require_torch
class TransfoXLModelLanguageGenerationTest(unittest.TestCase):
Expand Down
4 changes: 4 additions & 0 deletions tests/models/xglm/test_modeling_xglm.py
Original file line number Diff line number Diff line change
Expand Up @@ -357,6 +357,10 @@ def test_model_from_pretrained(self):
def test_model_parallelism(self):
super().test_model_parallelism()

@unittest.skip("This test is currently broken because of safetensors.")
def test_tf_from_pt_safetensors(self):
pass


@require_torch
class XGLMModelLanguageGenerationTest(unittest.TestCase):
Expand Down

0 comments on commit e107ae3

Please sign in to comment.