Skip to content

Commit 5be6ce4

Browse files
committed
Fix test case errors
1 parent 8cc7a94 commit 5be6ce4

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

qa/L0_simple_ensemble/ensemble_backpressure_test.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -138,26 +138,26 @@ def _run_inference(self, model_name, expected_responses_count=32):
138138
finally:
139139
triton_client.stop_stream()
140140

141-
def test_backpressure_limits_inflight(self):
141+
def test_max_inflight_requests_limit_4(self):
142142
"""
143143
Test that max_inflight_requests correctly limits concurrent
144144
responses.
145145
"""
146146
self._run_inference(model_name=MODEL_ENSEMBLE_LIMIT_4)
147147

148-
def test_backpressure_limit_one(self):
148+
def test_max_inflight_requests_limit_1(self):
149149
"""
150150
Test edge case: max_inflight_requests=1.
151151
"""
152152
self._run_inference(model_name=MODEL_ENSEMBLE_LIMIT_1)
153153

154-
def test_backpressure_disabled(self):
154+
def test_max_inflight_requests_limit_disabled(self):
155155
"""
156156
Test that an ensemble model without max_inflight_requests parameter works correctly.
157157
"""
158158
self._run_inference(model_name=MODEL_ENSEMBLE_DISABLED)
159159

160-
def test_backpressure_concurrent_requests(self):
160+
def test_max_inflight_requests_limit_concurrent_requests(self):
161161
"""
162162
Test that backpressure works correctly with multiple concurrent requests.
163163
Each request should have independent backpressure state.
@@ -178,7 +178,7 @@ def test_backpressure_concurrent_requests(self):
178178
for i in range(num_concurrent):
179179
clients[i].start_stream(callback=partial(callback, user_datas[i]))
180180
clients[i].async_stream_infer(
181-
model_name=MODEL_ENSEMBLE_ENABLED, inputs=inputs, outputs=outputs
181+
model_name=MODEL_ENSEMBLE_LIMIT_4, inputs=inputs, outputs=outputs
182182
)
183183

184184
# Collect and verify responses for all requests
@@ -203,7 +203,7 @@ def test_backpressure_concurrent_requests(self):
203203
for client in clients:
204204
client.stop_stream()
205205

206-
def test_backpressure_request_cancellation(self):
206+
def test_max_inflight_requests_limit_request_cancellation(self):
207207
"""
208208
Test that cancellation unblocks producers waiting on backpressure and that
209209
the client receives a cancellation error.
@@ -219,7 +219,7 @@ def test_backpressure_request_cancellation(self):
219219

220220
# Start the request
221221
triton_client.async_stream_infer(
222-
model_name=MODEL_ENSEMBLE_ENABLED, inputs=inputs, outputs=outputs
222+
model_name=MODEL_ENSEMBLE_LIMIT_4, inputs=inputs, outputs=outputs
223223
)
224224

225225
responses = []

0 commit comments

Comments
 (0)