ça c'est moi

pyomeca · Feb 7, 2025 · fc7fffb · fc7fffb
1 parent 6ec826e
commit fc7fffb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyorerun/biorbd_components/mesh.py b/pyorerun/biorbd_components/mesh.py
@@ -13,7 +13,9 @@ class TransformableMeshUpdater(Component):
     """
 
     def __init__(self, name: str, mesh: Trimesh, transform_callable: callable):
-        filename = mesh.metadata["file_name"] if "file_name" in mesh.metadata else mesh.metadata["header"].replace(" ", "")
+        filename = (
+            mesh.metadata["file_name"] if "file_name" in mesh.metadata else mesh.metadata["header"].replace(" ", "")
+        )
         self.__name = name + "/" + filename
         self.__mesh = mesh
 

diff --git a/pyorerun/biorbd_components/model_markers.py b/pyorerun/biorbd_components/model_markers.py
@@ -38,7 +38,7 @@ def compute_markers(self, q: np.ndarray) -> np.ndarray:
         nb_frames = q.shape[1]
         markers = np.zeros((3, self.nb_markers, nb_frames))
         for f in range(q.shape[1]):
-            markers[:, :, f] = self.callable_markers(q[:, f]).T
+            markers[:, :, f] = self.callable_markers(q[:, f]).squeeze().T
 
         return markers