hSrSSKJr SSKJr SSKJr SSKrSSK J r J r SSK J r Jr SSKJr SS KJrJrJr S /r"S S \\R,R.5rg) zFVector wrapper for converting between Array API compatible frameworks.) annotations) ModuleType)AnyN)ActTypeObsType) VectorEnv VectorWrapper) ArrayType)Devicearray_conversionmodule_name_to_namespaceArrayConversioncx\rSrSrSrS S SjjrS SjrSSS.SSjjrSrS r S r g)ra%Wraps a vector environment returning Array API compatible arrays so that it can be interacted with through a specific framework. Popular Array API frameworks include ``numpy``, ``torch``, ``jax.numpy``, ``cupy`` etc. With this wrapper, you can convert outputs from your environment to any of these frameworks. Conversely, actions are automatically mapped back to the environment framework, if possible without moving the data or device transfers. Notes: A vectorized version of :class:`gymnasium.wrappers.ArrayConversion` Example: >>> import gymnasium as gym # doctest: +SKIP >>> envs = gym.make_vec("JaxEnv-vx", 3) # doctest: +SKIP >>> envs = ArrayConversion(envs, xp=np) # doctest: +SKIP Nc[RRRU5 [R"X5 X lX0lX@lXPlg)atWrapper class to change inputs and outputs of environment to any Array API framework. Args: env: The Array API compatible environment to wrap env_xp: The Array API framework the environment is on target_xp: The Array API framework to convert to env_device: The device the environment is on target_device: The device on which Arrays should be returned N) gymutilsRecordConstructorArgs__init__r _env_xp _target_xp _env_device_target_device)selfenvenv_xp target_xp env_device target_devices d/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/wrappers/vector/array_conversion.pyrArrayConversion.__init__&s@" ''006t) #%+c [XRURS9nURR U5up#pEn[X R UR S9[X0R UR S9[X@R UR S9[XPR UR S9[X`R UR S94$)zTransforms the action to the specified xp module array type. Args: actions: The action to perform Returns: A tuple containing xp versions of the next observation, reward, termination, truncation, and extra info. xpdevice)r rrrsteprr)ractionsobsreward terminated truncatedinfos r r'ArrayConversion.step>s#7||DDTDTU3788==3I0ZD S__T=P=P Q V@S@S T t7J7J  Y??4CVCV W Tood>Q>Q R  r"seedoptionscU(a[X RURS9n[URR XS9UR UR S9$)zResets the environment returning xp-based observation and info. Args: seed: The seed for resetting the environment options: The options for resetting the environment, these are converted to xp arrays. Returns: xp-based observations and info r$r/)r rrrresetrr)rr0r1s r r3ArrayConversion.resetVsQ &LL1A1AG HHNNN 6&&  r"cURRRSS5nURRRSS5nURnUR nUUUUUR S.$)z5Returns the object pickle state with args and kwargs.zarray_api_compat.) env_xp_nametarget_xp_namerrr)r__name__replacerrrr)rr7r8rrs r __getstate__ArrayConversion.__getstate__psoll++334GL 1199:MrR%% ++ &,$*88   r"cUSUl[US5Ul[US5UlUSUlUSUlg)z%Sets the object pickle state using d.rr7r8rrN)rr rrrr)rds r __setstate__ArrayConversion.__setstate__~sIU8/-0@A 215E3FG\?0r")rrrrr)NN) rrrModuleType | strrrAr Device | NonerrB)r(rreturnz5tuple[ObsType, ArrayType, ArrayType, ArrayType, dict])r0zint | list[int] | Noner1zdict[str, Any] | NonerCztuple[ObsType, dict[str, Any]]) r9 __module__ __qualname____firstlineno____doc__rr'r3r;r?__static_attributes__r"r rrs (%)'+ , ,!,$ , " , % ,0  > 6(,)-  % '  (  4  1r")rG __future__rtypesrtypingr gymnasiumrgymnasium.corerrgymnasium.vectorrr gymnasium.vector.vector_envr #gymnasium.wrappers.array_conversionr r r __all__rrrrIr"r rSsKL"+51  n1mSYY%D%Dn1r"