added squeeze to get rid of excess dimension and fixed shape

BAMWelDX · Nov 3, 2024 · 737c133 · 737c133
1 parent 6ac88d3
commit 737c133
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/weldx/transformations/local_cs.py b/weldx/transformations/local_cs.py
@@ -748,11 +748,11 @@ def as_homogeneous_matrix(self, translation_unit: QuantityLike) -> np.ndarray:
         translation = np.resize(
             self.coordinates.data.to(translation_unit).m, (time_dim, 3)
         )
-        homogeneous_matrix = np.zeros((time_dim, 4, 4))
+        homogeneous_matrix = np.resize(np.identity(4), (time_dim, 4, 4))
         homogeneous_matrix[:, :3, :3] = rotation
         homogeneous_matrix[:, :3, 3] = translation
 
-        return homogeneous_matrix
+        return np.squeeze(homogeneous_matrix)
 
     def _interp_time_orientation(self, time: Time) -> xr.DataArray:
         """Interpolate the orientation in time."""