FIX ComplexWarning for numpy>=1.26 (#62)

gallantlab · Jun 21, 2024 · 42fc895 · 42fc895
1 parent fd22af6
commit 42fc895
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/himalaya/validation.py b/himalaya/validation.py
@@ -2,7 +2,10 @@
 import warnings
 
 import numpy as np
-from numpy.core.numeric import ComplexWarning
+try:
+    from numpy.core.numeric import ComplexWarning
+except ImportError:
+    from numpy.exceptions import ComplexWarning
 
 try:
     from scipy.sparse import issparse