From 238736393b226547d71146cf9c590a05a018e5b8 Mon Sep 17 00:00:00 2001
From: Google DeepMind <noreply@deepmind.com>
Date: Sun, 9 Jun 2024 20:55:40 -0700
Subject: [PATCH] #mjx Make `Data.time` strongly typed.

PiperOrigin-RevId: 641774798
Change-Id: I5e13e3cefeb7942083f623f29757a459656062dd
---
 mjx/mujoco/mjx/_src/io.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mjx/mujoco/mjx/_src/io.py b/mjx/mujoco/mjx/_src/io.py
index 564020f6..efa1da86 100644
--- a/mjx/mujoco/mjx/_src/io.py
+++ b/mjx/mujoco/mjx/_src/io.py
@@ -174,7 +174,7 @@ def make_data(m: Union[types.Model, mujoco.MjModel]) -> types.Data:
       nefc=nefc,
       ncon=ncon,
       solver_niter=jp.array(0, dtype=int),
-      time=jp.array(0.0),
+      time=jp.array(0.0, dtype=float),
       qpos=jp.array(m.qpos0),
       qvel=zero_nv,
       act=zero_na,