Skip to content

Commit 6f50bb9

Browse files
fix set_property of ModelDistributionPolicy with set type in python
1 parent 18e97b6 commit 6f50bb9

File tree

1 file changed

+8
-2
lines changed
  • src/bindings/python/src/pyopenvino/utils

1 file changed

+8
-2
lines changed

src/bindings/python/src/pyopenvino/utils/utils.cpp

+8-2
Original file line numberDiff line numberDiff line change
@@ -449,8 +449,14 @@ ov::Any py_object_to_any(const py::object& py_obj) {
449449
return py::cast<ov::intel_auto::SchedulePolicy>(py_obj);
450450
} else if (py::isinstance<ov::hint::SchedulingCoreType>(py_obj)) {
451451
return py::cast<ov::hint::SchedulingCoreType>(py_obj);
452-
} else if (py::isinstance<std::set<ov::hint::ModelDistributionPolicy>>(py_obj)) {
453-
return py::cast<std::set<ov::hint::ModelDistributionPolicy>>(py_obj);
452+
} else if (py::isinstance<py::set>(py_obj)) {
453+
std::set<ov::hint::ModelDistributionPolicy> model_set;
454+
for (auto item = py_obj.begin(); item != py_obj.end(); item++) {
455+
if (py::isinstance<ov::hint::ModelDistributionPolicy>(*item)) {
456+
model_set.insert(py::cast<ov::hint::ModelDistributionPolicy>(*item));
457+
}
458+
}
459+
return model_set;
454460
} else if (py::isinstance<ov::hint::ExecutionMode>(py_obj)) {
455461
return py::cast<ov::hint::ExecutionMode>(py_obj);
456462
} else if (py::isinstance<ov::log::Level>(py_obj)) {

0 commit comments

Comments
 (0)