diff --git a/source/pwrapper/numpyWrap.cpp b/source/pwrapper/numpyWrap.cpp index 9bec5103e..326be82b6 100644 --- a/source/pwrapper/numpyWrap.cpp +++ b/source/pwrapper/numpyWrap.cpp @@ -65,11 +65,11 @@ PyArrayContainer::ExtractData(void *_pParentPyArray) PyArrayObject *pParent = reinterpret_cast(pParentPyArray); int numDims = PyArray_NDIM(pParent); - long* pDims = (long*)PyArray_DIMS(pParent); + intptr_t* pDims = PyArray_DIMS(pParent); pData = PyArray_DATA(pParent); TotalSize = PyArray_SIZE(pParent); - Dims = std::vector(&pDims[0], &pDims[numDims]); + Dims = std::vector(pDims, pDims + numDims); int iDataType = PyArray_TYPE(pParent); switch(iDataType) { diff --git a/source/pwrapper/numpyWrap.h b/source/pwrapper/numpyWrap.h index 75907c441..a28cc2791 100644 --- a/source/pwrapper/numpyWrap.h +++ b/source/pwrapper/numpyWrap.h @@ -60,7 +60,7 @@ namespace Manta void *pData; NumpyTypes DataType; unsigned int TotalSize; - std::vector Dims; + std::vector Dims; private: void *pParentPyArray; };