added capsules in an attempt to prevent segmentation fault

Author: ParticularMiner

sparse_dot_topn/sparse_dot_topn.pyx

@@ -172,23 +172,24 @@ cpdef sparse_dot_free(
 
     sparse_dot_free_source(n_row, n_col, Ap, Aj, Ax, Bp, Bj, Bx, lower_bound, Cp, &vCj, &vCx)
 
-    cdef np.npy_intp nnz = Cp[n_row]
-    cdef np.ndarray[np.int32_t, ndim=1] c_indices = np.PyArray_SimpleNewFromData(1, &nnz, np.NPY_INT32, vCj.data())
+    cdef np.npy_intp nnz[1]
+    nnz[0] = Cp[n_row]
+    cdef np.ndarray[np.int32_t, ndim=1] c_indices = np.PyArray_SimpleNewFromData(4, &nnz[0], np.NPY_INT32, vCj.data())
     PyArray_ENABLEFLAGS(c_indices, np.NPY_OWNDATA)
-    cdef np.ndarray[np.double_t, ndim=1] c_data = np.PyArray_SimpleNewFromData(1, &nnz, np.NPY_DOUBLE, vCx.data())
+    cdef np.ndarray[np.double_t, ndim=1] c_data = np.PyArray_SimpleNewFromData(4, &nnz[0], np.NPY_DOUBLE, vCx.data())
     PyArray_ENABLEFLAGS(c_data, np.NPY_OWNDATA)
 
-    # cdef const char *name_vCj_capsule = "vCj"
-    # cdef int* vCj_data = vCj.data()
-    # vCj_capsule = PyCapsule_New(<void *> vCj_data, name_vCj_capsule, &free_ptr)
-    # if not PyCapsule_IsValid(vCj_capsule, name_vCj_capsule):
-        # raise ValueError(f"invalid pointer ({name_vCj_capsule}) to parameters")
-        #
-    # cdef const char *name_vCx_capsule = "vCx"
-    # cdef double* vCx_data = vCx.data()
-    # vCx_capsule = PyCapsule_New(<void *> vCx_data, name_vCx_capsule, &free_ptr)
-    # if not PyCapsule_IsValid(vCx_capsule, name_vCx_capsule):
-        # raise ValueError(f"invalid pointer ({name_vCx_capsule}) to parameters")
+    cdef const char *name_vCj_capsule = "vCj"
+    cdef int* vCj_data = vCj.data()
+    vCj_capsule = PyCapsule_New(<void *> vCj_data, name_vCj_capsule, &free_ptr)
+    if not PyCapsule_IsValid(vCj_capsule, name_vCj_capsule):
+        raise ValueError(f"invalid pointer ({name_vCj_capsule}) to parameters")
+        
+    cdef const char *name_vCx_capsule = "vCx"
+    cdef double* vCx_data = vCx.data()
+    vCx_capsule = PyCapsule_New(<void *> vCx_data, name_vCx_capsule, &free_ptr)
+    if not PyCapsule_IsValid(vCx_capsule, name_vCx_capsule):
+        raise ValueError(f"invalid pointer ({name_vCx_capsule}) to parameters")
 
     return c_indices, c_data