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

core test updated

Browse Source
This commit is contained in:
Mateusz Bencer 2022-03-04 11:44:10 +01:00
parent 959fef031a
commit 1a62074a75
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/bindings/python/tests/test_inference_engine/test_core.py
View File

@ -315,7 +315,8 @@ def test_add_extension_template_extension(device):
new_shapes = {"in_data": after_reshape} new_shapes = {"in_data": after_reshape}
assert model.input().partial_shape == before_reshape assert model.input().partial_shape == before_reshape
model.reshape(new_shapes) model.reshape(new_shapes)
assert model.input().partial_shape == after_reshape compiled = core.compile_model(model, device)
assert compiled.input().partial_shape == after_reshape
def test_add_extension(): def test_add_extension():