Doctest pls

Farama-Foundation · Dec 6, 2023 · acfeee2 · acfeee2
1 parent 756cbb3
commit acfeee2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gymnasium/envs/phys2d/cartpole.py b/gymnasium/envs/phys2d/cartpole.py
@@ -55,13 +55,13 @@ class CartPoleFunctional(
         >>> print(state)
         [ 0.46532142 -0.27484107  0.13302994 -0.20361817]
         >>> print(env.transition(state, 0, params=params))
-        [ 0.4598246  -0.6357784   0.12895757  0.12780523]
+        [ 0.4598246  -0.6357784   0.12895757  0.1278053]
         >>> env.transform(jax.jit)
         >>> state = env.initial(key, params=params)
         >>> print(state)
         [ 0.46532142 -0.27484107  0.13302994 -0.20361817]
-        >>> print(env.transition(state, 0, params=params))  # doctest: +SKIP
-        [ 0.4598246  -0.6357784   0.12895757  0.12780523]
+        >>> print(env.transition(state, 0, params=params))
+        [ 0.4598246  -0.6357784   0.12895757  0.1278053]
         >>> vkey = jax.random.split(key, 10)
         >>> env.transform(jax.vmap)
         >>> vstate = env.initial(vkey, params=params)