Skip to content

Commit 8492549

Browse files
committed
Update error msg checks
1 parent 62184db commit 8492549

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

qa/L0_input_validation/input_validation_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ def get_input_array(input_size, np_dtype):
195195
triton_client.infer(model_name=model_name, inputs=inputs)
196196
err_str = str(e.exception)
197197
self.assertIn(
198-
f"expected {input_size} string elements for inference input 'INPUT1', got {input_size-2}",
198+
f"expected {input_size} string elements for inference input 'INPUT1' for model '{model_name}', got {input_size-2}",
199199
err_str,
200200
)
201201

@@ -208,7 +208,7 @@ def get_input_array(input_size, np_dtype):
208208
triton_client.infer(model_name=model_name, inputs=inputs)
209209
err_str = str(e.exception)
210210
self.assertIn(
211-
f"expected {input_size} string elements for inference input 'INPUT1', got {input_size+2}",
211+
f"expected {input_size} string elements for inference input 'INPUT1' for model '{model_name}', got {input_size+2}",
212212
err_str,
213213
)
214214

0 commit comments

Comments
 (0)