Skip to content

Commit 254af43

Browse files
committed
Attempt to fix CI
1 parent 5b1f07d commit 254af43

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

newton/tests/test_benchmark_simulation.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -223,17 +223,16 @@ def __exit__(self, exc_type, exc_value, traceback):
223223

224224
def test_kamino_contact_capacity_scales_with_world_count(self):
225225
"""Scale the model-wide Kamino contact cap across replicated worlds."""
226+
from newton._src.solvers.kamino.examples.rl.simulation import RigidBodySim # noqa: PLC0415
227+
226228
settings = SimpleNamespace(
227229
collision_detector=SimpleNamespace(max_contacts=1000),
228230
solver=SimpleNamespace(collision_detector=None, dynamics=SimpleNamespace(linear_solver_type=None)),
229231
)
230232
model = SimpleNamespace(world_count=4096)
231233

232234
with (
233-
patch(
234-
"newton._src.solvers.kamino.examples.rl.simulation.RigidBodySim.default_settings",
235-
return_value=settings,
236-
),
235+
patch.object(RigidBodySim, "default_settings", return_value=settings),
237236
patch("benchmark_kamino.newton.solvers.SolverKamino", return_value=object()) as solver_kamino,
238237
):
239238
solver = DRLegsBenchmarkWorkload.create_solver(model, sim_dt=0.001)

0 commit comments

Comments
 (0)