Function get_edge_labels

graphlearn/python/c/py_wrapper.h:344–353

}

PyObject* get_edge_labels(LookupEdgesResponse* res) {
  npy_intp shape[1];
  shape[0] = res->Size();
  PyArray_Descr* descr = PyArray_DescrFromType(NPY_INT32);
  PyObject* obj = PyArray_Zeros(1, shape, descr, 0);
  PyArrayObject* np_array = reinterpret_cast<PyArrayObject*>(obj);
  memcpy(PyArray_DATA(np_array), res->Labels(),
         res->Size() * INT32_BYTES);
  return obj;
}

PyObject* get_edge_int_attributes(LookupEdgesResponse* res) {
  return get_int_attributes(res);