diff --git a/tests/exporters/test_export.py b/tests/exporters/test_export.py index 3e5b96301..9ce117176 100644 --- a/tests/exporters/test_export.py +++ b/tests/exporters/test_export.py @@ -157,6 +157,7 @@ def test_export(self, test_name, name, model_name, task, neuron_config_construct @is_inferentia_test @require_vision @require_sentence_transformers + @requires_neuronx def test_export_sentence_transformers(self, test_name, name, model_name, task, neuron_config_constructor): self._neuronx_export(test_name, name, model_name, task, neuron_config_constructor)