Skip to content

Commit 06d1ddd

Browse files
committed
Update Python API to new signature
1 parent ef309c0 commit 06d1ddd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

apis/python/src/tiledb/vector_search/module.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,12 +98,12 @@ static void declare_kmeans_query(py::module& m, const std::string& suffix) {
9898
const ColMajorMatrix<float>& centroids,
9999
const ColMajorMatrix<float>& query_vectors,
100100
std::vector<uint64_t>& indices,
101-
std::vector<shuffled_ids_type>& ids,
101+
std::vector<uint64_t>& ids,
102102
size_t nprobe,
103103
size_t k_nn,
104104
bool nth,
105105
size_t nthreads) -> ColMajorMatrix<size_t> { // TODO change return type
106-
auto r = detail::ivf::qv_query_heap_infinite_ram<T>(
106+
auto r = detail::ivf::qv_query_heap_infinite_ram(
107107
parts,
108108
centroids,
109109
query_vectors,

0 commit comments

Comments
 (0)