fix: fromiter() to make array from map()

engnadeau · Aug 26, 2022 · c3bbeff · c3bbeff
1 parent 3aff2e5
commit c3bbeff
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pybotics/optimization.py b/pybotics/optimization.py
@@ -123,7 +123,9 @@ def compute_absolute_errors(
     :param positions: Array of Cartesian positions, shape=(n-poses, 3)
     :param robot: Robot model
     """
-    result = np.array(map(compute_absolute_error, qs, positions, repeat(robot)))
+    result = np.fromiter(
+        map(compute_absolute_error, qs, positions, repeat(robot)), dtype=np.float64
+    )
     return result
 
 
@@ -155,4 +157,7 @@ def compute_relative_errors(
     robot: Robot,
 ) -> npt.NDArray[np.float64]:
     """Compute the relative errors of a given set of position combinations."""
-    return np.array(map(compute_relative_error, qs_a, qs_b, distances, repeat(robot)))
+    return np.fromiter(
+        map(compute_relative_error, qs_a, qs_b, distances, repeat(robot)),
+        dtype=np.float64,
+    )