python: Add more generic check

stack-of-tasks · Dec 13, 2024 · 1e4df2b · 1e4df2b
1 parent dc53def
commit 1e4df2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bindings/python/pinocchio/visualize/meshcat_visualizer.py b/bindings/python/pinocchio/visualize/meshcat_visualizer.py
@@ -73,7 +73,7 @@ def applyScalingOnHomegeneousTransform(
     homogeneous_transform: np.ndarray, scale: np.ndarray
 ) -> np.ndarray:
     assert homogeneous_transform.shape == (4, 4)
-    assert scale.shape == (3,)
+    assert scale.size == 3
     scale = np.array(scale).flatten()
     S = np.diag(np.concatenate((scale, [1.0])))
     return homogeneous_transform @ S