Skip to content

GitLab

Unverified Commit 632ff3c3 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[FlaxSpeechEncoderDecoderModel] Skip from_encoder_decoder_pretrained (#16236) 

* skip the test

* fix

* fix skip
parent b6e06c84
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 0 deletions
tests/speech_encoder_decoder/test_modeling_flax_speech_encoder_decoder.py
View file @ 632ff3c3
......@@ -691,6 +691,10 @@ class FlaxWav2Vec2GPT2ModelTest(FlaxEncoderDecoderMixin, unittest.TestCase):
self.assertEqual(len(fx_outputs), len(pt_outputs_loaded), "Output lengths differ between Flax and PyTorch")
self.assert_almost_equals(fx_logits, pt_logits_loaded.numpy(), 4e-2)
@unittest.skip("Re-enable this test once this issue is fixed: https://github.com/google/jax/issues/9941")
def test_encoder_decoder_model_from_encoder_decoder_pretrained(self):
pass
@require_flax
class FlaxWav2Vec2BartModelTest(FlaxEncoderDecoderMixin, unittest.TestCase):
......@@ -807,3 +811,7 @@ class FlaxWav2Vec2BartModelTest(FlaxEncoderDecoderMixin, unittest.TestCase):
self.assertEqual(len(fx_outputs), len(pt_outputs_loaded), "Output lengths differ between Flax and PyTorch")
self.assert_almost_equals(fx_logits, pt_logits_loaded.numpy(), 4e-2)
@unittest.skip("Re-enable this test once this issue is fixed: https://github.com/google/jax/issues/9941")
def test_encoder_decoder_model_from_encoder_decoder_pretrained(self):
pass
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment