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

radiocosmology · Sep 26, 2024 · a606465 · a606465
1 parent d5cb01a
commit a606465
Showing 1 changed file with 3 additions and 0 deletions.
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