Skip to content

Commit 015c473

Browse files
committed
clean up numpy includes and target numpy 2.0 C API
1 parent 676c2fd commit 015c473

File tree

4 files changed

+13
-17
lines changed

4 files changed

+13
-17
lines changed

stringdtype/stringdtype/src/casts.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
11
#ifndef _NPY_CASTS_H
22
#define _NPY_CASTS_H
33

4+
// needed for Py_UCS4
45
#include <Python.h>
56

7+
// need these defines and includes for PyArrayMethod_Spec
68
#define PY_ARRAY_UNIQUE_SYMBOL stringdtype_ARRAY_API
7-
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
9+
#define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
10+
#define NPY_TARGET_VERSION NPY_2_0_API_VERSION
811
#define NO_IMPORT_ARRAY
9-
#include "numpy/arrayobject.h"
1012
#include "numpy/experimental_dtype_api.h"
11-
#include "numpy/halffloat.h"
12-
#include "numpy/ndarraytypes.h"
1313

1414
PyArrayMethod_Spec **
1515
get_casts();

stringdtype/stringdtype/src/dtype.h

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,15 @@
77
// clang-format on
88

99
#define PY_ARRAY_UNIQUE_SYMBOL stringdtype_ARRAY_API
10-
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
10+
#define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
11+
#define NPY_TARGET_VERSION NPY_2_0_API_VERSION
1112
#define NO_IMPORT_ARRAY
1213
#include "numpy/arrayobject.h"
1314
#include "numpy/experimental_dtype_api.h"
15+
#include "numpy/halffloat.h"
1416
#include "numpy/ndarraytypes.h"
1517
#include "numpy/npy_math.h"
18+
#include "numpy/ufuncobject.h"
1619

1720
typedef struct {
1821
PyArray_Descr base;

stringdtype/stringdtype/src/main.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
#include <Python.h>
22

33
#define PY_ARRAY_UNIQUE_SYMBOL stringdtype_ARRAY_API
4-
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
4+
#define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
5+
#define NPY_TARGET_VERSION NPY_2_0_API_VERSION
56
#include "numpy/arrayobject.h"
67
#include "numpy/experimental_dtype_api.h"
78

@@ -88,9 +89,8 @@ static struct PyModuleDef moduledef = {
8889
PyMODINIT_FUNC
8990
PyInit__main(void)
9091
{
91-
if (_import_array() < 0) {
92-
return NULL;
93-
}
92+
import_array();
93+
9494
if (import_experimental_dtype_api(13) < 0) {
9595
return NULL;
9696
}

stringdtype/stringdtype/src/umath.c

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,10 @@
11
#include <Python.h>
22

3-
#define PY_ARRAY_UNIQUE_SYMBOL stringdtype_ARRAY_API
4-
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
5-
#define NO_IMPORT_ARRAY
6-
#include "numpy/arrayobject.h"
7-
#include "numpy/experimental_dtype_api.h"
8-
#include "numpy/ndarraytypes.h"
9-
#include "numpy/ufuncobject.h"
3+
#include "umath.h"
104

115
#include "dtype.h"
126
#include "static_string.h"
137
#include "string.h"
14-
#include "umath.h"
158

169
static NPY_CASTING
1710
multiply_resolve_descriptors(

0 commit comments

Comments
 (0)