Skip to content

Commit 1f96abd

Browse files
committed
Update error msg
1 parent 962d1f0 commit 1f96abd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

qa/L0_input_validation/input_validation_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def test_client_input_shape_validation(self):
169169
triton_client.infer(model_name=model_name, inputs=inputs)
170170
err_str = str(e.exception)
171171
self.assertIn(
172-
f"'INPUT0' got unexpected elements count 16, expected 8",
172+
f"input 'INPUT0' got unexpected elements count 16, expected 8",
173173
err_str,
174174
)
175175

@@ -202,7 +202,7 @@ def identity_inference(triton_client, np_array, binary_data):
202202
triton_client.infer(model_name=model_name, inputs=inputs)
203203
err_str = str(e.exception)
204204
self.assertIn(
205-
f"'INPUT0' got unexpected elements count 16, expected 8",
205+
f"input 'INPUT0' got unexpected elements count 16, expected 8",
206206
err_str,
207207
)
208208

@@ -286,7 +286,7 @@ def test_client_input_shm_size_validation(self):
286286
triton_client.infer(model_name=model_name, inputs=inputs)
287287
err_str = str(e.exception)
288288
self.assertIn(
289-
f"'INPUT0' got unexpected byte size {input_byte_size+4}, expected {input_byte_size}",
289+
f"input 'INPUT0' got unexpected byte size {input_byte_size+4}, expected {input_byte_size}",
290290
err_str,
291291
)
292292

@@ -300,7 +300,7 @@ def test_client_input_shm_size_validation(self):
300300
triton_client.infer(model_name=model_name, inputs=inputs)
301301
err_str = str(e.exception)
302302
self.assertIn(
303-
f"'INPUT1' got unexpected byte size {input_byte_size-4}, expected {input_byte_size}",
303+
f"input 'INPUT1' got unexpected byte size {input_byte_size-4}, expected {input_byte_size}",
304304
err_str,
305305
)
306306

0 commit comments

Comments
 (0)