natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PyOV] Fix passing of the key in data dispatcher (#15941)

This commit is contained in:
Jan Iwaszkiewicz 2023-02-24 20:07:59 +01:00 committed by GitHub
parent 57cb7015f0
commit 6c0e2686ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/bindings/python/src/openvino/runtime/utils/data_helpers/data_dispatcher.py
View File

@ -212,7 +212,7 @@ def update_tensor(
key: Optional[ValidKeys] = None,
) -> None:
if hasattr(inputs, "__array__"):
update_tensor(normalize_arrays(inputs, is_shared=False), request, key=None)
update_tensor(normalize_arrays(inputs, is_shared=False), request, key)
return None
raise TypeError(f"Incompatible inputs of type: {type(inputs)} under {key} key!")

5
src/bindings/python/tests/test_runtime/test_infer_request.py
View File

@ -925,6 +925,7 @@ def test_array_like_input_request(device, shared_flag):
request, _, input_data = abs_model_with_data(device, Type.f32, np.single)
model_input_object = ArrayLikeObject(input_data.tolist())
model_input_list = [ArrayLikeObject(input_data.tolist())]
model_input_dict = {0: ArrayLikeObject(input_data.tolist())}
# Test single array-like object in InferRequest().Infer()
res_object = request.infer(model_input_object, shared_memory=shared_flag)
@ -934,6 +935,10 @@ def test_array_like_input_request(device, shared_flag):
res_list = request.infer(model_input_list)
assert np.array_equal(res_list[request.model_outputs[0]], np.abs(input_data))
# Test dict of array-like objects to use normalize_inputs()
res_dict = request.infer(model_input_dict)
assert np.array_equal(res_dict[request.model_outputs[0]], np.abs(input_data))
@pytest.mark.parametrize("shared_flag", [True, False])
def test_array_like_input_async(device, shared_flag):