From 92c161a61ba1d0529eb7b2d38f61298b9ddbf92e Mon Sep 17 00:00:00 2001 From: Baruch Tabanpour Date: Thu, 1 Feb 2024 13:37:28 -0800 Subject: [PATCH] Add ncon to __init__. PiperOrigin-RevId: 603468404 Change-Id: I20c622b8f6f84ecacb9f3cdd26ac9560cf2788be --- mjx/mujoco/mjx/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mjx/mujoco/mjx/__init__.py b/mjx/mujoco/mjx/__init__.py index 8656e618..d3e048d2 100644 --- a/mjx/mujoco/mjx/__init__.py +++ b/mjx/mujoco/mjx/__init__.py @@ -16,6 +16,7 @@ # pylint:disable=g-importing-member from mujoco.mjx._src.collision_driver import collision +from mujoco.mjx._src.collision_driver import ncon from mujoco.mjx._src.constraint import count_constraints from mujoco.mjx._src.constraint import make_constraint from mujoco.mjx._src.device import device_get_into @@ -43,7 +44,7 @@ from mujoco.mjx._src.smooth import kinematics from mujoco.mjx._src.smooth import rne from mujoco.mjx._src.smooth import transmission from mujoco.mjx._src.solver import solve -from mujoco.mjx._src.support import is_sparse from mujoco.mjx._src.support import full_m +from mujoco.mjx._src.support import is_sparse from mujoco.mjx._src.support import mul_m from mujoco.mjx._src.types import *