Skip to content

Commit

Permalink
Fix misleading line for creating list of models in rollout_test.py
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 717582208
Change-Id: I1b4e40fc5ccbfe68fb7dc240715c704f1f68d9fe
  • Loading branch information
yuvaltassa authored and copybara-github committed Jan 20, 2025
1 parent 699a676 commit ff47834
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python/mujoco/rollout_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
"""tests for rollout function."""

import concurrent.futures
import copy
import threading

from absl.testing import absltest
Expand All @@ -23,6 +24,7 @@
from mujoco import rollout
import numpy as np


# -------------------------- models used for testing ---------------------------

TEST_XML = r"""
Expand Down Expand Up @@ -473,7 +475,7 @@ def test_threading(self):
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(
Expand Down

0 comments on commit ff47834

Please sign in to comment.