Merge pull request #1438 from compas-dev/fix-numpy2-float-import

Fix numpy error

Author: tomvanmele

CHANGELOG.md

@@ -18,6 +18,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 * Fixed publish to YAK via CI workflow.
 * Added selector for `test` and `prod` to CI workflow.
+* Fixed `AttributeError` in `compas.data.DataEncoder.default` due to `np.float_` no longer being available in `numpy>=2`.
 
 ### Removed
 

src/compas/data/encoders.py

@@ -30,6 +30,11 @@
 try:
     import numpy as np
 
+    try:
+        np_float = np.float_
+    except AttributeError:
+        np_float = np.float64
+
     numpy_support = True
 except (ImportError, SyntaxError):
     numpy_support = False
@@ -143,7 +148,7 @@ def default(self, o):
                 ),  # type: ignore
             ):
                 return int(o)
-            if isinstance(o, (np.float_, np.float16, np.float32, np.float64)):  # type: ignore
+            if isinstance(o, (np_float, np.float16, np.float32, np.float64)):  # type: ignore
                 return float(o)
             if isinstance(o, np.bool_):
                 return bool(o)

tests/compas/data/test_json_numpy.py

@@ -15,3 +15,24 @@ def test_json_numpy():
 
 except ImportError:
     pass
+
+
+try:
+    import numpy as np
+
+    try:
+        np_float = np.float_
+    except AttributeError:
+        np_float = np.float64
+
+    def test_json_numpy_float():
+        before = [
+            np.array([1, 2, 3], dtype=np_float),
+            np.array([1.0, 2.0, 3.0]),
+            np_float(1.0),
+        ]
+        after = compas.json_loads(compas.json_dumps(before))
+        assert after == [[1, 2, 3], [1.0, 2.0, 3.0], 1.0]
+
+except ImportError:
+    pass