Remove deprecated tests

rtzr · Aug 29, 2023 · 0d343d7 · 0d343d7
1 parent 07f572e
commit 0d343d7
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/test_model_call.py b/tests/test_model_call.py
@@ -37,11 +37,7 @@ def test_with_input_name(protocol_and_port):
 
     client = InferenceClient.create_with(MODEL_NAME, f"{TRITON_HOST}:{port}", protocol=protocol)
 
-    sample = np.random.rand(1, 100).astype(np.float32)
-    result = client({client.input_name_list[0]: sample})
-    print(f"Result: {np.isclose(result, sample).all()}")
-
     sample = np.random.rand(100, 100).astype(np.float32)
-    result = client({client.default_model_spec.input_name[0]: sample})
+    result = client({client.default_model_spec.model_input[0].name: sample})
 
     print(f"Result: {np.isclose(result, sample).all()}")