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

[PyOV] Fix docs and add test for Model._get_raw_address (#20109)

Browse Source
This commit is contained in:
Anastasia Kuporosova 2023-09-28 14:06:20 +02:00 committed by GitHub
parent 9a88eed9f2
commit 744f5d4130
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/bindings/python/src/pyopenvino/graph/model.cpp
View File

@ -1004,7 +1004,9 @@ void regclass_graph_Model(py::module m) {
return reinterpret_cast<uint64_t>(&self); return reinterpret_cast<uint64_t>(&self);
}, },
R"( R"(
Returns raw address of the Model object. Returns a raw address of the Model object from C++.
Use this function in order to compare underlying C++ addresses instead of using `__eq__` in Python.
:return: a raw address of the Model object. :return: a raw address of the Model object.
:rtype: int :rtype: int

9
src/bindings/python/tests/test_runtime/test_model.py
View File

@ -657,3 +657,12 @@ def test_model_add_remove_result_parameter_sink():
assert ["Assign"] == [sink.get_type_name() for sink in assign_nodes] assert ["Assign"] == [sink.get_type_name() for sink in assign_nodes]
model.remove_sink(assign) model.remove_sink(assign)
assert len(model.sinks) == 0 assert len(model.sinks) == 0
def test_model_get_raw_address():
model = generate_add_model()
model_with_same_addr = model
model_different = generate_add_model()
assert model._get_raw_address() == model_with_same_addr._get_raw_address()
assert model._get_raw_address() != model_different._get_raw_address()