abandon inf1

huggingface · Jan 15, 2024 · 2e04b86 · 2e04b86
1 parent db0cba8
commit 2e04b86
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)