chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix binding issue with scalar attributes.

Browse Source 
PiperOrigin-RevId: 720136044
Change-Id: I0846a93ef36a23371d1bd504adff02dcce0abcf9
This commit is contained in:
Alessio Quaglino
2025-01-27 05:46:36 -08:00
committed by Copybara-Service
parent d86900aac1
commit d21f624264
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/_src/support.py
+1 -1
View File
@@ -367,7 +367,7 @@ class BindModel(object):
self.id = ids
def __getattr__(self, name: str):
return getattr(self.model, self.prefix + name)[self.id, :]
return getattr(self.model, self.prefix + name)[self.id, ...]
def _bind_model(self: Model, obj: Sequence[Any]) -> BindModel:
mjx/mujoco/mjx/_src/support_test.py
+1
View File
@@ -219,6 +219,7 @@ class SupportTest(parameterized.TestCase):
)
np.testing.assert_array_equal(mx.bind(s.joints).axis, m.jnt_axis)
np.testing.assert_array_equal(mx.bind(s.joints).qposadr, m.jnt_qposadr)
for i in range(m.njnt):
np.testing.assert_array_equal(m.bind(s.joints[i]).axis, m.jnt_axis[i, :])
np.testing.assert_array_equal(mx.bind(s.joints[i]).axis, m.jnt_axis[i, :])