Modify mjx.Model tex_data type.

PiperOrigin-RevId: 828606695 Change-Id: I708dcf8e75766bf16d339999a9e8c70c8f65fd0c
2025-11-05 13:31:57 -08:00
parent c34ac712a0
commit d0f32a7fe6
2 changed files with 9 additions and 1 deletions
@@ -819,7 +819,7 @@ class Model(PyTreeNode):
  tex_width: np.ndarray
  tex_nchannel: np.ndarray
  tex_adr: np.ndarray
-  tex_data: jax.Array
+  tex_data: np.ndarray
  mat_rgba: jax.Array
  mat_texid: np.ndarray
  pair_dim: np.ndarray