Skip to content

Commit 0d343d7

Browse files
committed
Remove deprecated tests
1 parent 07f572e commit 0d343d7

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/test_model_call.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,11 +37,7 @@ def test_with_input_name(protocol_and_port):
3737

3838
client = InferenceClient.create_with(MODEL_NAME, f"{TRITON_HOST}:{port}", protocol=protocol)
3939

40-
sample = np.random.rand(1, 100).astype(np.float32)
41-
result = client({client.input_name_list[0]: sample})
42-
print(f"Result: {np.isclose(result, sample).all()}")
43-
4440
sample = np.random.rand(100, 100).astype(np.float32)
45-
result = client({client.default_model_spec.input_name[0]: sample})
41+
result = client({client.default_model_spec.model_input[0].name: sample})
4642

4743
print(f"Result: {np.isclose(result, sample).all()}")

0 commit comments

Comments
 (0)