fix version string parsing

mscheltienne · mscheltienne · commit 275618e38ae6 · 2024-08-27T14:33:22.000+02:00
diff --git a/quantities/dimensionality.py b/quantities/dimensionality.py
@@ -9,7 +9,8 @@
 from .registry import unit_registry
 from .decorators import memoize
 
-_np_version = tuple(map(int, np.__version__.split('.')))
+_np_version = tuple(map(int, np.__version__.split(".dev")[0].split(".")))
+
 
 def assert_isinstance(obj, types):
     try:
diff --git a/quantities/quantity.py b/quantities/quantity.py
@@ -291,7 +291,7 @@ def __array_prepare__(self, obj, context=None):
         return res
 
     def __array_wrap__(self, obj, context=None, return_scalar=False):
-        _np_version = tuple(map(int, np.__version__.split('.')))
+        _np_version = tuple(map(int, np.__version__.split(".dev")[0].split(".")))
         # For NumPy < 2.0 we do old behavior
         if _np_version < (2, 0, 0):
             if not isinstance(obj, Quantity):
diff --git a/quantities/tests/test_umath.py b/quantities/tests/test_umath.py
@@ -3,7 +3,7 @@
 from .. import units as pq
 from .common import TestCase, unittest
 
-_np_version = tuple(map(int, np.__version__.split('.')))
+_np_version = tuple(map(int, np.__version__.split(".dev")[0].split(".")))
 
 
 class TestUmath(TestCase):
