diff --git a/ttnn/cpp/pybind11/tensor.cpp b/ttnn/cpp/pybind11/tensor.cpp index e2f6f35800b..2d00d4ee2e3 100644 --- a/ttnn/cpp/pybind11/tensor.cpp +++ b/ttnn/cpp/pybind11/tensor.cpp @@ -256,6 +256,7 @@ void tensor_mem_config_module(py::module& m_tensor) { auto pyCoreRangeSet = static_cast>(m_tensor.attr("CoreRangeSet")); pyCoreRangeSet.def(py::init<>([](const std::set& core_ranges) { return CoreRangeSet(core_ranges); })) + .def(py::init<>([](const std::vector& core_ranges) { return CoreRangeSet(tt::stl::Span(core_ranges)); })) .def( "bounding_box", &CoreRangeSet::bounding_box,