tmp

huggingface · Dec 12, 2024 · 633fefc · 633fefc
1 parent d700dc2
commit 633fefc
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -1382,7 +1382,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("question-answering")
         question = "Whats my name?"
@@ -1567,7 +1566,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("fill-mask")
         text = "The capital of France is [MASK]."
@@ -1947,7 +1945,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("token-classification")
         text = "My Name is Philipp and i live in Germany."
@@ -2109,7 +2106,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("feature-extraction")
         text = "My Name is Philipp and i live in Germany."
@@ -2358,7 +2354,6 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
     SPEEDUP_CACHE = 1.1
 
     @parameterized.expand([(False,), (True,)])
-    @pytest.mark.run_in_series
     # TODO: still gotta find out why this needs to be ran in series / why it fails in parallel
     # my guess is that the model surgery is happening in parallel and that's causing the issue
     def test_inference_old_onnx_model(self, use_cache):
@@ -2967,7 +2962,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("image-classification")
         url = "http://images.cocodataset.org/val2017/000000039769.jpg"
@@ -3134,7 +3128,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("image-segmentation")
         url = "http://images.cocodataset.org/val2017/000000039769.jpg"
@@ -3318,7 +3311,6 @@ def test_pipeline_ort_model(self, model_arch):
 
         gc.collect()
 
-    @pytest.mark.run_in_series
     def test_pipeline_model_is_none(self):
         pipe = pipeline("audio-classification")
         data = self._generate_random_audio_data()
@@ -3701,7 +3693,6 @@ def _get_onnx_model_dir(self, model_id, model_arch, test_name):
 
         return onnx_model_dir
 
-    @pytest.mark.run_in_series
     def test_inference_old_onnx_model(self):
         tokenizer = get_preprocessor("t5-small")
         model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")