@@ -203,6 +203,8 @@ def test_implicit(self, jacobian, actuation, spring, damper):
203203 "opt.jacobian" : jacobian ,
204204 "opt.disableflags" : DisableBit .CONTACT | actuation | spring | damper ,
205205 "opt.integrator" : IntegratorType .IMPLICITFAST ,
206+ # TODO(team): remove override when mujoco warp feature matches mujoco
207+ "opt.enableflags" : EnableBit .INVDISCRETE ,
206208 },
207209 )
208210
@@ -491,7 +493,11 @@ def _getattr(arr):
491493 integrator = (IntegratorType .EULER , IntegratorType .IMPLICITFAST , IntegratorType .RK4 ),
492494 )
493495 def test_step2 (self , xml , integrator ):
494- mjm , mjd , m , _ = test_data .fixture (xml , qvel_noise = 0.01 , ctrl_noise = 0.1 , overrides = {"opt.integrator" : integrator })
496+ # TODO(team): remove enableflags override when mujoco warp feature matches mujoco
497+ enableflags = EnableBit .INVDISCRETE if integrator == IntegratorType .IMPLICITFAST else 0
498+ mjm , mjd , m , _ = test_data .fixture (
499+ xml , qvel_noise = 0.01 , ctrl_noise = 0.1 , overrides = {"opt.integrator" : integrator , "opt.enableflags" : enableflags }
500+ )
495501
496502 # some of the fields updated by step2
497503 step2_field = [
0 commit comments