Function get_edge_weights

graphlearn/python/c/py_wrapper.h:333–342

}

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

PyObject* get_edge_labels(LookupEdgesResponse* res) {
  npy_intp shape[1];