From acfeee2dffc37826e50c1d3d5d1a3461adf7818b Mon Sep 17 00:00:00 2001 From: ariel Date: Wed, 6 Dec 2023 10:58:36 -0500 Subject: [PATCH] Doctest pls --- gymnasium/envs/phys2d/cartpole.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gymnasium/envs/phys2d/cartpole.py b/gymnasium/envs/phys2d/cartpole.py index 6308c8ddc..b33b651ee 100644 --- 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)