From fb84ee4e177896f4e00b547addffcc9cb6136caa Mon Sep 17 00:00:00 2001
From: Google DeepMind <noreply@deepmind.com>
Date: Mon, 17 Feb 2025 09:22:59 -0800
Subject: [PATCH] #mjx Always define `prefix` attribute in `Bind*` classes.

PiperOrigin-RevId: 727885838
Change-Id: I3e7dd0541fecf1eb913bb98b30e3df146ea665b8
---
 mjx/mujoco/mjx/_src/support.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mjx/mujoco/mjx/_src/support.py b/mjx/mujoco/mjx/_src/support.py
index bcb83c64..06fb4670 100644
--- a/mjx/mujoco/mjx/_src/support.py
+++ b/mjx/mujoco/mjx/_src/support.py
@@ -291,6 +291,7 @@ class BindModel(object):
 
   def __init__(self, model: Model, specs: Sequence[mujoco.MjStruct]):
     self.model = model
+    self.prefix = ''
     try:
       iter(specs)
     except TypeError:
@@ -386,6 +387,7 @@ class BindData(object):
   ):
     self.data = data
     self.model = model
+    self.prefix = ''
     try:
       iter(specs)
     except TypeError:
@@ -393,7 +395,6 @@ class BindData(object):
     ids = []
     for spec in specs:
       if isinstance(spec, mujoco.MjsBody):
-        self.prefix = ''
         idx = name2id(model, mujoco.mjtObj.mjOBJ_BODY, spec.name)
       elif isinstance(spec, mujoco.MjsJoint):
         self.prefix = 'jnt_'