feat(weighted_median): call 'numpy.import_array()' after cimport

radiocosmology · Sep 26, 2024 · 95ca0d2 · 95ca0d2
1 parent d5cb01a
commit 95ca0d2
Show file tree

Hide file tree

Showing 2 changed files with 6 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
+np.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,9 @@ from MedianTree cimport Tree, Data
 cimport numpy as np
 cimport cython
 
+# Required for numpy 2.0 compatibility
+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