We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 07f572e commit 0d343d7Copy full SHA for 0d343d7
tests/test_model_call.py
@@ -37,11 +37,7 @@ def test_with_input_name(protocol_and_port):
37
38
client = InferenceClient.create_with(MODEL_NAME, f"{TRITON_HOST}:{port}", protocol=protocol)
39
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
-
44
sample = np.random.rand(100, 100).astype(np.float32)
45
- result = client({client.default_model_spec.input_name[0]: sample})
+ result = client({client.default_model_spec.model_input[0].name: sample})
46
47
print(f"Result: {np.isclose(result, sample).all()}")
0 commit comments