From ff4783482c0c9fafb43da275dd143eb92cb29d64 Mon Sep 17 00:00:00 2001
From: Yuval Tassa <tassa@google.com>
Date: Mon, 20 Jan 2025 11:12:24 -0800
Subject: [PATCH] Fix misleading line for creating list of models in
 rollout_test.py

PiperOrigin-RevId: 717582208
Change-Id: I1b4e40fc5ccbfe68fb7dc240715c704f1f68d9fe
---
 python/mujoco/rollout_test.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/mujoco/rollout_test.py b/python/mujoco/rollout_test.py
index af8a5d3a..25689a68 100644
--- a/python/mujoco/rollout_test.py
+++ b/python/mujoco/rollout_test.py
@@ -15,6 +15,7 @@
 """tests for rollout function."""
 
 import concurrent.futures
+import copy
 import threading
 
 from absl.testing import absltest
@@ -23,6 +24,7 @@ import mujoco
 from mujoco import rollout
 import numpy as np
 
+
 # -------------------------- models used for testing ---------------------------
 
 TEST_XML = r"""
@@ -473,7 +475,7 @@ class MuJoCoRolloutTest(parameterized.TestCase):
     def thread_initializer():
       thread_local.data = mujoco.MjData(model)
 
-    model_list = [model] * nroll
+    model_list = [copy.copy(model) for _ in range(nroll)]
 
     def call_rollout(initial_state, control, state, sensordata):
       rollout.rollout(