skip continuous batching test (#1567)

neuralmagic · Jan 29, 2024 · e29211c · e29211c
1 parent 10a9b38
commit e29211c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/tests/deepsparse/schedulers/test_continuous_batching_scheduler.py b/tests/deepsparse/schedulers/test_continuous_batching_scheduler.py
@@ -16,10 +16,12 @@
 
 import numpy
 
+import pytest
 from deepsparse.operators import EngineOperator
 from deepsparse.schedulers import ContinuousBatchingScheduler
 
 
+@pytest.mark.skip("skip continuous batching tests")
 def test_continuous_batching_executor_thread():
     # simple test that ContinuousBatchingScheduler can be instantiated and return
     # a result from a request, for testing multi-batch execution, making enough

diff --git a/tests/deepsparse/schedulers/utils/test_continuous_batching_executor.py b/tests/deepsparse/schedulers/utils/test_continuous_batching_executor.py
@@ -16,13 +16,15 @@
 
 import numpy
 
+import pytest
 from deepsparse.operators import EngineOperator
 from deepsparse.schedulers.utils import (
     ContinuousBatchingExecutorThread,
     ContinuousBatchingQueues,
 )
 
 
+@pytest.mark.skip("skip continuous batching tests")
 def test_continuous_batching_executor_thread():
     # mobilenet model with batch_size=2
     engine_operator = EngineOperator("zoo:mobilenet_v2-1.0-imagenet-base")

diff --git a/tests/deepsparse/transformers/text_generation/integration_tests/test_llms.py b/tests/deepsparse/transformers/text_generation/integration_tests/test_llms.py
@@ -132,6 +132,7 @@ def setup(self, params_dict, max_new_tokens, internal_kv_cache):
         self.default_pipeline = None
         self.max_new_tokens = max_new_tokens
 
+    @pytest.mark.skip("skip continuous batching tests")
     def test_continuous_batching_pipeline(self, setup):
 
         pipeline = self.get_pipeline(