BUG: Fix import of AxisError for older versions of numpy.

WarrenWeckesser · Jul 29, 2024 · e32ef5f · e32ef5f
1 parent 6b3ba8d
commit e32ef5f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ufunclab/_wrapped.py b/ufunclab/_wrapped.py
@@ -4,7 +4,10 @@
 
 import operator
 import numpy as np
-from numpy.exceptions import AxisError
+try:
+    from numpy.exceptions import AxisError
+except ImportError:
+    from numpy import AxisError
 from ufunclab._convert_to_base import convert_to_base as _convert_to_base