Skip to content

Commit

Permalink
SKIP
Browse files Browse the repository at this point in the history
  • Loading branch information
RedTachyon committed Dec 6, 2023
1 parent fb968b8 commit a9aea84
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions gymnasium/envs/phys2d/cartpole.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit a9aea84

Please sign in to comment.