core test updated

2022-03-04 11:44:10 +01:00 · 2022-03-04 11:44:10 +01:00 · 1a62074a75
commit 1a62074a75
parent 959fef031a
1 changed files with 2 additions and 1 deletions
--- a/src/bindings/python/tests/test_inference_engine/test_core.py
+++ b/src/bindings/python/tests/test_inference_engine/test_core.py
@ -315,7 +315,8 @@ def test_add_extension_template_extension(device):
    new_shapes = {"in_data": after_reshape}
    assert model.input().partial_shape == before_reshape
    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():