diff --git a/libs/numpy/src/numpy.cpp b/libs/numpy/src/numpy.cpp index 883d9716..0cc3f5f6 100644 --- a/libs/numpy/src/numpy.cpp +++ b/libs/numpy/src/numpy.cpp @@ -12,9 +12,19 @@ namespace boost namespace numpy { +#if PY_MAJOR_VERSION == 2 +static void wrap_import_array() { + import_array(); +} +#else +static void * wrap_import_array() { + import_array(); +} +#endif + void initialize(bool register_scalar_converters) { - import_array(); + wrap_import_array(); import_ufunc(); if (register_scalar_converters) dtype::register_scalar_converters();