From 0c68574aa786bf555635ed29cbaf8ad431e9f4a9 Mon Sep 17 00:00:00 2001 From: Mateusz Tabaka Date: Fri, 10 Dec 2021 12:01:44 +0100 Subject: [PATCH] Fix compilation error in pyopenvino (#9150) --- src/bindings/python/src/pyopenvino/pyopenvino.cpp | 6 +++--- .../tests/test_inference_engine/test_function.py | 10 +++++----- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/bindings/python/src/pyopenvino/pyopenvino.cpp b/src/bindings/python/src/pyopenvino/pyopenvino.cpp index e4e7d31fbd1..90c961e2f55 100644 --- a/src/bindings/python/src/pyopenvino/pyopenvino.cpp +++ b/src/bindings/python/src/pyopenvino/pyopenvino.cpp @@ -70,10 +70,10 @@ PYBIND11_MODULE(pyopenvino, m) { m.def("set_batch", &ov::set_batch); m.def( "set_batch", - [](const std::shared_ptr& f, int64_t value) { - return ov::set_batch(f, ov::Dimension(value)); + [](const std::shared_ptr& model, int64_t value) { + return ov::set_batch(model, ov::Dimension(value)); }, - py::arg("function"), + py::arg("model"), py::arg("batch_size") = -1); regclass_graph_PyRTMap(m); diff --git a/src/bindings/python/tests/test_inference_engine/test_function.py b/src/bindings/python/tests/test_inference_engine/test_function.py index 7dce9cfbf41..d97e10cc079 100644 --- a/src/bindings/python/tests/test_inference_engine/test_function.py +++ b/src/bindings/python/tests/test_inference_engine/test_function.py @@ -277,7 +277,7 @@ def test_get_batch(): param1 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data1") param2 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data2") add = ops.add(param1, param2) - func = Function(add, [param1, param2], "TestFunction") + func = Model(add, [param1, param2], "TestFunction") param = func.get_parameters()[0] param.set_layout(Layout("NC")) assert get_batch(func) == 2 @@ -289,7 +289,7 @@ def test_get_batch_CHWN(): param3 = ops.parameter(Shape([3, 1, 3, 4]), dtype=np.float32, name="data3") add = ops.add(param1, param2) add2 = ops.add(add, param3) - func = Function(add2, [param1, param2, param3], "TestFunction") + func = Model(add2, [param1, param2, param3], "TestFunction") param = func.get_parameters()[0] param.set_layout(Layout("CHWN")) assert get_batch(func) == 4 @@ -299,7 +299,7 @@ def test_set_batch_dimension(): param1 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data1") param2 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data2") add = ops.add(param1, param2) - func = Function(add, [param1, param2], "TestFunction") + func = Model(add, [param1, param2], "TestFunction") func_param1 = func.get_parameters()[0] func_param2 = func.get_parameters()[1] # batch == 2 @@ -318,7 +318,7 @@ def test_set_batch_int(): param1 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data1") param2 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data2") add = ops.add(param1, param2) - func = Function(add, [param1, param2], "TestFunction") + func = Model(add, [param1, param2], "TestFunction") func_param1 = func.get_parameters()[0] func_param2 = func.get_parameters()[1] # batch == 2 @@ -337,7 +337,7 @@ def test_set_batch_default_batch_size(): param1 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data1") param2 = ops.parameter(Shape([2, 1]), dtype=np.float32, name="data2") add = ops.add(param1, param2) - func = Function(add, [param1, param2], "TestFunction") + func = Model(add, [param1, param2], "TestFunction") func_param1 = func.get_parameters()[0] func_param1.set_layout(Layout("NC")) set_batch(func)