Explicitly set the numpy type in _write_array (#2128)

kounelisagis · web-flow · commit a6f4fdc23ab2 · 2025-01-03T14:58:05.000-05:00
`QueryExperimental::set_data_buffer` requires an unsigned integer as its last argument. It seems that with numpy versions earlier than 2, this argument becomes a numpy float64 (with `buffer_sizes[i]` being a numpy uint64 and `ncells` being a Python int). Let's fix it by explicitly cast to numpy uint64. Closes #2127, closes #2124.
diff --git a/tiledb/array.py b/tiledb/array.py
@@ -899,7 +899,7 @@ def _write_array(
             q.set_data_buffer(
                 buffer_name,
                 output_values[i],
-                buffer_sizes[i] * ncells,
+                np.uint64(buffer_sizes[i] * ncells),
             )
 
             # Set offsets buffer

Original file line number	Diff line number	Diff line change
`@@ -899,7 +899,7 @@ def _write_array(`
`899`	`899`	`q.set_data_buffer(`
`900`	`900`	`buffer_name,`
`901`	`901`	`output_values[i],`
`902`		`- buffer_sizes[i] * ncells,`
	`902`	`+ np.uint64(buffer_sizes[i] * ncells),`
`903`	`903`	`)`
`904`	`904`
`905`	`905`	`# Set offsets buffer`