From a901e4307169e11d83bfbbfc321b09bb47d10bc2 Mon Sep 17 00:00:00 2001 From: Filippo Luca Ferretti Date: Thu, 14 Nov 2024 13:07:57 +0100 Subject: [PATCH] Speed up `js.link.idxs_to_names` --- src/jaxsim/api/link.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/jaxsim/api/link.py b/src/jaxsim/api/link.py index 25da8f74c..23b4d3732 100644 --- a/src/jaxsim/api/link.py +++ b/src/jaxsim/api/link.py @@ -4,6 +4,7 @@ import jax import jax.numpy as jnp import jax.scipy.linalg +import numpy as np import jaxsim.api as js import jaxsim.rbda @@ -96,7 +97,7 @@ def idxs_to_names( The names of the links. """ - return tuple(idx_to_name(model=model, link_index=idx) for idx in link_indices) + return tuple(np.array(model.kin_dyn_parameters.link_names)[list(link_indices)]) # =========