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(