RYDE-WORK/pybind11
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/pybind11.git synced 2026-02-02 21:45:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed py:array constructor from failing for complex types

Browse Source 
The array(const buffer_info &info) constructor fails when given
complex types since their format string is 'Zd' or 'Zf' which has
a length of two and causes an error here:

if (info.format.size() != 1)
    throw std::runtime_error("Unsupported buffer format!");

Fixed by allowing format sizes of one and two.
This commit is contained in:
Jan Dohl 2015-12-15 04:16:49 +01:00
parent 61d67f0462
commit ab92eb3765
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/pybind11/numpy.h
View File

@ -96,7 +96,7 @@ public:
array(const buffer_info &info) { array(const buffer_info &info) {
API& api = lookup_api(); API& api = lookup_api();
if (info.format.size() != 1) if ((info.format.size() < 1) || (info.format.size() > 2))
throw std::runtime_error("Unsupported buffer format!"); throw std::runtime_error("Unsupported buffer format!");
int fmt = (int) info.format[0]; int fmt = (int) info.format[0];
if (info.format == "Zd") if (info.format == "Zd")