Fix misleading line for creating list of models in rollout_test.py

PiperOrigin-RevId: 717582208
Change-Id: I1b4e40fc5ccbfe68fb7dc240715c704f1f68d9fe
This commit is contained in:
Yuval Tassa
2025-01-20 11:12:24 -08:00
committed by Copybara-Service
parent 699a6765e3
commit ff4783482c
+3 -1
View File
@@ -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(