restore t5 tests

huggingface · Nov 4, 2024 · 8fff6bf · 8fff6bf
1 parent 5ec92d8
commit 8fff6bf
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 27 deletions.
diff --git a/tests/cli/test_export_cli.py b/tests/cli/test_export_cli.py
@@ -303,9 +303,6 @@ def test_replace_unet(self):
                 check=True,
             )
 
-    @unittest.skip(
-        "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-    )
     @requires_neuronx
     def test_encoder_decoder(self):
         model_id = "hf-internal-testing/tiny-random-t5"
@@ -335,9 +332,6 @@ def test_encoder_decoder(self):
                 check=True,
             )
 
-    @unittest.skip(
-        "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-    )
     @requires_neuronx
     def test_encoder_decoder_optional_outputs(self):
         model_id = "hf-internal-testing/tiny-random-t5"
@@ -369,9 +363,6 @@ def test_encoder_decoder_optional_outputs(self):
                 check=True,
             )
 
-    @unittest.skip(
-        "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-    )
     @requires_neuronx
     def test_encoder_decoder_tp2(self):
         model_id = "michaelbenayoun/t5-tiny-random"

diff --git a/tests/exporters/test_export.py b/tests/exporters/test_export.py
@@ -310,9 +310,6 @@ def test_export_sd_with_fused_lora_weights(self):
             )
 
 
-@unittest.skip(
-    "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-)
 @is_inferentia_test
 @requires_neuronx
 class NeuronEncoderDecoderExportTestCase(unittest.TestCase):

diff --git a/tests/generation/test_generate.py b/tests/generation/test_generate.py
@@ -35,9 +35,6 @@ def _test_model_generation_trn(model, tokenizer, batch_size, input_length, **gen
     assert sample_output.shape[0] == batch_size
 
 
-@pytest.mark.skip(
-    "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-)
 @is_inferentia_test
 @requires_neuronx
 def test_seq2seq_generation_beam(neuron_seq2seq_beam_path):
@@ -58,9 +55,6 @@ def test_seq2seq_generation_beam(neuron_seq2seq_beam_path):
     assert len(output[0].unique()) <= 5 + 1  # +1 for `decoder_start_token_id`
 
 
-@pytest.mark.skip(
-    "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-)
 @is_inferentia_test
 @requires_neuronx
 def test_seq2seq_generation_beam_with_optional_outputs(neuron_seq2seq_beam_path_with_optional_outputs):
@@ -83,9 +77,6 @@ def test_seq2seq_generation_beam_with_optional_outputs(neuron_seq2seq_beam_path_
     assert "decoder_hidden_states" in output
 
 
-@pytest.mark.skip(
-    "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-)
 @is_inferentia_test
 @requires_neuronx
 def test_seq2seq_generation_greedy(neuron_seq2seq_greedy_path):
@@ -106,9 +97,6 @@ def test_seq2seq_generation_greedy(neuron_seq2seq_greedy_path):
     assert len(output[0]) <= 5 + 1  # +1 for `decoder_start_token_id`
 
 
-@pytest.mark.skip(
-    "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-)
 @is_inferentia_test
 @requires_neuronx
 def test_seq2seq_generation_greedy_with_optional_outputs(neuron_seq2seq_greedy_path_with_optional_outputs):
@@ -129,9 +117,6 @@ def test_seq2seq_generation_greedy_with_optional_outputs(neuron_seq2seq_greedy_p
     assert "decoder_hidden_states" in output
 
 
-@pytest.mark.skip(
-    "T5 compilation broken since neuron sdk 2.20, wait for the fix: https://github.com/aws-neuron/aws-neuron-sdk/issues/1013."
-)
 @is_inferentia_test
 @requires_neuronx
 def test_seq2seq_generation_tp2(neuron_seq2seq_tp2_path):