diff --git a/python/cuvs/cuvs/neighbors/cagra/cagra.pyx b/python/cuvs/cuvs/neighbors/cagra/cagra.pyx index 4cc7e4c1b..f1c0c9af5 100644 --- a/python/cuvs/cuvs/neighbors/cagra/cagra.pyx +++ b/python/cuvs/cuvs/neighbors/cagra/cagra.pyx @@ -468,16 +468,13 @@ def search(SearchParams search_params, exp_rows=n_queries, exp_cols=k) cdef cagraSearchParams* params = &search_params.params - cdef cydlpack.DLManagedTensor queries_dlpack = \ - cydlpack.dlpack_c(queries_cai) - cdef cydlpack.DLManagedTensor neighbors_dlpack = \ - cydlpack.dlpack_c(neighbors_cai) - cdef cydlpack.DLManagedTensor distances_dlpack = \ - cydlpack.dlpack_c(distances_cai) + cdef cydlpack.DLManagedTensor queries_dlpack = cydlpack.dlpack_c(queries_cai) + cdef cydlpack.DLManagedTensor neighbors_dlpack = cydlpack.dlpack_c(neighbors_cai) + cdef cydlpack.DLManagedTensor distances_dlpack = cydlpack.dlpack_c(distances_cai) with cuda_interruptible(): cagraSearch( - resources_, + deref(resources_), params, index.index, &queries_dlpack,