Skip to content

Commit e29211c

Browse files
authored
skip continuous batching test (#1567)
1 parent 10a9b38 commit e29211c

File tree

3 files changed

+5
-0
lines changed

3 files changed

+5
-0
lines changed

tests/deepsparse/schedulers/test_continuous_batching_scheduler.py

+2
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,12 @@
1616

1717
import numpy
1818

19+
import pytest
1920
from deepsparse.operators import EngineOperator
2021
from deepsparse.schedulers import ContinuousBatchingScheduler
2122

2223

24+
@pytest.mark.skip("skip continuous batching tests")
2325
def test_continuous_batching_executor_thread():
2426
# simple test that ContinuousBatchingScheduler can be instantiated and return
2527
# a result from a request, for testing multi-batch execution, making enough

tests/deepsparse/schedulers/utils/test_continuous_batching_executor.py

+2
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,15 @@
1616

1717
import numpy
1818

19+
import pytest
1920
from deepsparse.operators import EngineOperator
2021
from deepsparse.schedulers.utils import (
2122
ContinuousBatchingExecutorThread,
2223
ContinuousBatchingQueues,
2324
)
2425

2526

27+
@pytest.mark.skip("skip continuous batching tests")
2628
def test_continuous_batching_executor_thread():
2729
# mobilenet model with batch_size=2
2830
engine_operator = EngineOperator("zoo:mobilenet_v2-1.0-imagenet-base")

tests/deepsparse/transformers/text_generation/integration_tests/test_llms.py

+1
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@ def setup(self, params_dict, max_new_tokens, internal_kv_cache):
132132
self.default_pipeline = None
133133
self.max_new_tokens = max_new_tokens
134134

135+
@pytest.mark.skip("skip continuous batching tests")
135136
def test_continuous_batching_pipeline(self, setup):
136137

137138
pipeline = self.get_pipeline(

0 commit comments

Comments
 (0)