Fix: properly handle Py_None reference count. (#272)

Use Py_RETURN_NONE to return Py_None
alibaba · Aug 2, 2023 · 526e7d1 · 526e7d1
1 parent e4b5090
commit 526e7d1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/graphlearn/python/c/py_wrapper.h b/graphlearn/python/c/py_wrapper.h
@@ -678,7 +678,7 @@ PyObject* get_dag_value(GetDagValuesResponse* res,
   auto t = res->GetValue(node_id, key);
   auto values = std::get<0>(t);
   if (values == nullptr) {
-    return Py_None;
+    Py_RETURN_NONE;
   }
   int32_t size = values->Size();
   npy_intp shape[1];
@@ -713,11 +713,11 @@ PyObject* get_dag_value_indice(GetDagValuesResponse* res,
   auto t = res->GetValue(node_id, key);
   auto indices = std::get<1>(t);
   if (indices == nullptr) {
-    return Py_None;
+    Py_RETURN_NONE;
   }
   int32_t size = indices->Size();
   if (size == 0) {
-    return Py_None;
+    Py_RETURN_NONE;
   }
   npy_intp shape[1];
   shape[0] = size;