fix(cython): fix numpy 2.0 cimport compatibility

radiocosmology · Jun 17, 2024 · f548670 · f548670
1 parent 4ef9b0b
commit f548670
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/caput/truncate.pyx b/caput/truncate.pyx
@@ -8,6 +8,9 @@ from cython.parallel import prange
 import numpy as np
 cimport numpy as cnp
 
+# Required for numpy >= 2.0 compatibility
+cnp.import_array()
+
 cdef extern from "truncate.hpp":
     inline int bit_truncate(int val, int err) nogil
 

diff --git a/caput/weighted_median.pyx b/caput/weighted_median.pyx
@@ -12,6 +12,10 @@ from MedianTree cimport Tree, Data
 cimport numpy as np
 cimport cython
 
+# Required for numpy >= 2.0
+# TODO: not sure if there should be some sort of conditional here
+np.import_array()
+
 # Define the fused types that can be used for the data or weights in the median routine
 ctypedef fused data_t:
     int