Fix update to numpy 2.0.0

sodascience · Jun 17, 2024 · ba9a1f8 · ba9a1f8
1 parent 6dcb730
commit ba9a1f8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/metasyncontrib/disclosure/utils.py b/metasyncontrib/disclosure/utils.py
@@ -6,7 +6,11 @@
 
 import numpy as np
 import polars as pl
-from numpy.core._exceptions import UFuncTypeError
+
+try:
+    from numpy.core._exceptions import UFuncTypeError
+except ImportError:
+    from numpy._core._exceptions import UFuncTypeError
 
 
 def _compute_dominance(block_values, reverse=False):