hphSrSSKJr SSKJr SSKrSSKJr SSK J r SSK J r S/r "SS\ 5rg) z4Vector wrapper for converting between NumPy and Jax.) annotationsN)DependencyNotInstalled) VectorEnv)ArrayConversion JaxToNumpyc0^\rSrSrSrSU4SjjrSrU=r$)ra;Wraps a jax vector environment so that it can be interacted with through numpy arrays. Notes: A vectorized version of :class:`gymnasium.wrappers.JaxToNumpy` Actions must be provided as numpy arrays and observations, rewards, terminations and truncations will be returned as numpy arrays. Example: >>> import gymnasium as gym # doctest: +SKIP >>> envs = gym.make_vec("JaxEnv-vx", 3) # doctest: +SKIP >>> envs = JaxToNumpy(envs) # doctest: +SKIP cX>[c [S5e[TU] U[[S9 g)zzWraps an environment such that the input and outputs are numpy arrays. Args: env: the vector jax environment to wrap Nz8Jax is not installed, run `pip install "gymnasium[jax]"`)env_xp target_xp)jnprsuper__init__np)selfenv __class__s `/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/wrappers/vector/jax_to_numpy.pyrJaxToNumpy.__init__s0 ;(J  SB7)rr)__name__ __module__ __qualname____firstlineno____doc__r__static_attributes__ __classcell__)rs@rrrs  8 8r)r __future__r jax.numpynumpyr rgymnasium.errorrgymnasium.vectorr*gymnasium.wrappers.vector.array_conversionr__all__rrrrr&s/:"2&F .88r