diff options
| -rw-r--r-- | numpy/numarray/_capi.c | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/numpy/numarray/_capi.c b/numpy/numarray/_capi.c index 534057095..1387404b7 100644 --- a/numpy/numarray/_capi.c +++ b/numpy/numarray/_capi.c @@ -3402,14 +3402,18 @@ PyMODINIT_FUNC init_capi(void) _Error = PyErr_NewException("numpy.numarray._capi.error", NULL, NULL); /* Create a CObject containing the API pointer array's address */ -#if PY_VERSION_HEX >= 0x02070000 +#if PY_VERSION_HEX >= 0x03010000 m = PyModule_Create(&moduledef); +#else + m = Py_InitModule("_capi", _libnumarrayMethods); +#endif + +#if PY_VERSION_HEX >= 0x02070000 c_api_object = PyCapsule_New((void *)libnumarray_API, NULL, NULL); if (c_api_object == NULL) { PyErr_Clear(); } #else - m = Py_InitModule("_capi", _libnumarrayMethods); c_api_object = PyCObject_FromVoidPtr((void *)libnumarray_API, NULL); #endif |