Skip to content

Commit d150159

Browse files
committed
Apply suggestions from review
1 parent 9f36b54 commit d150159

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

include/pybind11/detail/class.h

+2-1
Original file line numberDiff line numberDiff line change
@@ -602,9 +602,9 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla
602602
return -1;
603603
}
604604
view->obj = obj;
605+
view->ndim = 1; // See discussion on PR #5407.
605606
view->internal = info;
606607
view->buf = info->ptr;
607-
view->ndim = (int) info->ndim;
608608
view->itemsize = info->itemsize;
609609
view->len = view->itemsize;
610610
for (auto s : info->shape) {
@@ -615,6 +615,7 @@ extern "C" inline int pybind11_getbuffer(PyObject *obj, Py_buffer *view, int fla
615615
view->format = const_cast<char *>(info->format.c_str());
616616
}
617617
if ((flags & PyBUF_ND) == PyBUF_ND) {
618+
view->ndim = (int) info->ndim;
618619
view->shape = info->shape.data();
619620
}
620621
if ((flags & PyBUF_STRIDES) == PyBUF_STRIDES) {

tests/test_buffers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def test_to_pybuffer():
247247
info = m.get_py_buffer(mat, m.PyBUF_SIMPLE)
248248
assert info.itemsize == ctypes.sizeof(ctypes.c_float)
249249
assert info.len == mat.rows() * mat.cols() * info.itemsize
250-
assert info.ndim == 2
250+
assert info.ndim == 1 # See discussion on PR #5407.
251251
assert info.shape is None
252252
assert info.strides is None
253253
assert info.suboffsets is None

0 commit comments

Comments
 (0)