weighted

radiocosmology · Aug 9, 2024 · c947af4 · c947af4
1 parent 7a95cf3
commit c947af4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/caput/weighted_median.pyx b/caput/weighted_median.pyx
@@ -10,9 +10,11 @@ from libcpp.deque cimport deque
 from cython.parallel import prange, parallel
 from MedianTree cimport Tree, Data
 cimport numpy as np
-np.import_array()
 cimport cython
 
+# Required for numpy 2.0 compatibility
+np.PyArray_ImportNumPyAPI()
+
 # Define the fused types that can be used for the data or weights in the median routine
 ctypedef fused data_t:
     int