diff --git a/gymnasium/envs/phys2d/cartpole.py b/gymnasium/envs/phys2d/cartpole.py index 4fff609c4..9f8cfbcc6 100644 --- a/gymnasium/envs/phys2d/cartpole.py +++ b/gymnasium/envs/phys2d/cartpole.py @@ -54,13 +54,13 @@ class CartPoleFunctional( >>> state = env.initial(key, params=params) >>> print(state) [ 0.46532142 -0.27484107 0.13302994 -0.20361817] - >>> print(env.transition(state, 0, params=params)) + >>> print(env.transition(state, 0, params=params)) # doctest: +SKI [ 0.4598246 -0.6357784 0.12895757 0.1278053 ] >>> env.transform(jax.jit) >>> state = env.initial(key, params=params) - >>> print(state) + >>> print(state) # doctest: +SKI [ 0.46532142 -0.27484107 0.13302994 -0.20361817] - >>> print(env.transition(state, 0, params=params)) + >>> print(env.transition(state, 0, params=params)) # doctest: +SKIP [ 0.4598246 -0.6357784 0.12895757 0.1278053 ] >>> vkey = jax.random.split(key, 10) >>> env.transform(jax.vmap)