From baf1c43b3cee0aa171f63112d733fa539e1bf5f5 Mon Sep 17 00:00:00 2001 From: Yuval Tassa Date: Mon, 20 Jan 2025 14:26:33 -0800 Subject: [PATCH] Fix more misleading line for creating list of models in rollout_test.py PiperOrigin-RevId: 717628943 Change-Id: I83a314565ce4ec683afeea18c8c8ce9e23d5452e --- python/mujoco/rollout_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/mujoco/rollout_test.py b/python/mujoco/rollout_test.py index 25689a68..23139ba8 100644 --- a/python/mujoco/rollout_test.py +++ b/python/mujoco/rollout_test.py @@ -532,7 +532,7 @@ class MuJoCoRolloutTest(parameterized.TestCase): sensordata = np.empty((nroll, nstep, model.nsensordata)) control = np.random.randn(nroll, nstep, model.nu) - model_list = [model] * nroll + model_list = [copy.copy(model) for _ in range(nroll)] data_list = [mujoco.MjData(model) for _ in range(num_workers)] rollout.rollout( @@ -561,7 +561,7 @@ class MuJoCoRolloutTest(parameterized.TestCase): sensordata = np.empty((nroll, nstep, model.nsensordata)) control = np.random.randn(nroll, nstep, model.nu) - model_list = [model] * nroll + model_list = [copy.copy(model) for _ in range(nroll)] data_list = [mujoco.MjData(model) for _ in range(num_workers)] with rollout.Rollout(nthread=num_workers) as rollout_: @@ -610,7 +610,7 @@ class MuJoCoRolloutTest(parameterized.TestCase): sensordata = np.empty((nroll, nstep, model.nsensordata)) control = np.random.randn(nroll, nstep, model.nu) - model_list = [model] * nroll + model_list = [copy.copy(model) for _ in range(nroll)] data_list = [mujoco.MjData(model) for _ in range(num_workers)] for _ in range(2): @@ -871,7 +871,7 @@ def py_rollout( nstep = control.shape[1] if isinstance(model, mujoco.MjModel): - model = [model] * nroll + model = [copy.copy(model) for _ in range(nroll)] nstate = mujoco.mj_stateSize(model[0], mujoco.mjtState.mjSTATE_FULLPHYSICS)