Function get_sampling_node_ids

graphlearn/python/c/py_wrapper.h:420–430

}

PyObject* get_sampling_node_ids(SamplingResponse* res) {
  int32_t size = res->TotalNeighborCount();
  npy_intp shape[1];
  shape[0] = size;
  PyArray_Descr* descr = PyArray_DescrFromType(NPY_INT64);
  PyObject* obj = PyArray_Zeros(1, shape, descr, 0);
  PyArrayObject* np_array = reinterpret_cast<PyArrayObject*>(obj);
  memcpy(reinterpret_cast<int64_t*>(PyArray_DATA(np_array)),
         res->GetNeighborIds(), size * INT64_BYTES);
  return obj;
}

PyObject* get_sampling_edge_ids(SamplingResponse* res) {
  int32_t size = res->TotalNeighborCount();