Wrap fragment_timestamp_range + fixes

TileDB-Inc · Dec 5, 2024 · 944161a · 944161a
1 parent c0e126a
commit 944161a
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/tiledb/cc/query.cc b/tiledb/cc/query.cc
@@ -62,6 +62,8 @@ void init_query(py::module &m) {
 
       .def("fragment_uri", &Query::fragment_uri)
 
+      .def("fragment_timestamp_range", &Query::fragment_timestamp_range)
+
       .def("query_status", &Query::query_status)
 
       .def("set_condition", &Query::set_condition)
@@ -71,13 +73,13 @@ void init_query(py::module &m) {
       //     uint64_t))&Query::set_data_buffer);
 
       .def("set_data_buffer",
-           [](Query &q, std::string name, py::array a, uint32_t buff_size) {
-             q.set_data_buffer(name, const_cast<void *>(a.data()), buff_size);
+           [](Query &q, std::string name, py::array a, uint64_t nelements) {
+             QueryExperimental::set_data_buffer(q, name, const_cast<void *>(a.data()), nelements);
            })
 
       .def("set_offsets_buffer",
-           [](Query &q, std::string name, py::array a, uint32_t buff_size) {
-             q.set_offsets_buffer(name, (uint64_t *)(a.data()), buff_size);
+           [](Query &q, std::string name, py::array a, uint64_t nelements) {
+             q.set_offsets_buffer(name, (uint64_t *)(a.data()), nelements);
            })
 
       .def("set_subarray",
@@ -86,8 +88,8 @@ void init_query(py::module &m) {
            })
 
       .def("set_validity_buffer",
-           [](Query &q, std::string name, py::array a, uint32_t buff_size) {
-             q.set_validity_buffer(name, (uint8_t *)(a.data()), buff_size);
+           [](Query &q, std::string name, py::array a, uint64_t nelements) {
+             q.set_validity_buffer(name, (uint8_t *)(a.data()), nelements);
            })
 
       .def("_submit", &Query::submit, py::call_guard<py::gil_scoped_release>())