Fix import for numpy 2.0

brian-team · Sep 7, 2023 · 2e5c326 · 2e5c326
1 parent a7c8e35
commit 2e5c326
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/brian2/units/fundamentalunits.py b/brian2/units/fundamentalunits.py
@@ -23,7 +23,12 @@
 from warnings import warn
 
 import numpy as np
-from numpy import VisibleDeprecationWarning
+
+try:
+    from numpy.exceptions import VisibleDeprecationWarning  # numpy 2.x
+except ImportError:
+    from numpy import VisibleDeprecationWarning  # numpy 1.x
+
 from sympy import latex
 
 __all__ = [