Function get_node_set

graphlearn/python/c/py_wrapper.h:509–518

}

PyObject* get_node_set(SubGraphResponse* res) {
  npy_intp shape[1];
  shape[0] = res->NodeCount();
  PyArray_Descr* descr = PyArray_DescrFromType(NPY_INT64);
  PyObject* obj = PyArray_Zeros(1, shape, descr, 0);
  PyArrayObject* np_array = reinterpret_cast<PyArrayObject*>(obj);
  memcpy(PyArray_DATA(np_array), res->NodeIds(),
         res->NodeCount() * INT64_BYTES);
  return obj;
}

PyObject* get_row_idx(SubGraphResponse* res) {
  npy_intp shape[1];