hSrSSKJr SSKJrJr SSKrSSKJ r \ \\ \R4\\\R4\\ \R4\\ \44r\ \\ \R4\\\R4\\ \R4\\ \R4\\ \44rS S SjjrS S SjjrS S Sjjrg)zLContains methods for step compatibility, from old-to-new and new-to-old API.) annotations) SupportsFloatUnionN)ObsTypec *[U5S:XaU$[U5S:XdeUup#pEUSLa0URSS5nUUU=(a U(+U=(a UU4$[U[5(az[R "UVs/sHowRSS5PM sn5nUU[R "U[R"U55[R "XF5U4$[U[5(ax[U5nURS[R"U[S95nUU[R "U[R"U55[R "XF5U4$[S[U535es snf)aZFunction to transform step returns to new step API irrespective of input API. .. py:currentmodule:: gymnasium.Env Args: step_returns (tuple): Items returned by :meth:`step`. Can be ``(obs, rew, done, info)`` or ``(obs, rew, terminated, truncated, info)`` is_vector_env (bool): Whether the ``step_returns`` are from a vector environment FTimeLimit.truncated)dtypefUnexpected value of infos, as is_vector_envs=False, expects `info` to be a list or dict, actual type: ) lenpop isinstancelistnparray logical_and logical_notdictzerosbool TypeErrortype) step_returns is_vector_env observationsrewardsdonesinfos truncatedinfonum_envss `/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/utils/step_api_compatibility.py(convert_to_terminated_truncated_step_apir$s <A< A%%%.:+ u E ! "7?I'i-#)  t $ $DIJED/7EJIubnnY&?@u0  t $ $5zH "7(RV9WXIubnnY&?@u0  xy}DzEyFG )KsFc[U5S:XaU$[U5S:XdeUup#pEnUSLa0U(dU(aU=(a U(+US'UUU=(d UU4$[U[5(aS[XeU5H)upxn U(d U (dMU=(a U (+US'M+ UU[R "XE5U4$[U[ 5(a[R "[R"U5[R"U55(a.[R"U[R"U55US'UU[R "XE5U4$[S[U535e)aZFunction to transform step returns to old step API irrespective of input API. .. py:currentmodule:: gymnasium.Env Args: step_returns (tuple): Items returned by :meth:`step`. Can be ``(obs, rew, done, info)`` or ``(obs, rew, terminated, truncated, info)`` is_vector_env (bool): Whether the ``step_returns`` are from a vector environment r rFr r ) r rrzipr logical_orranyrrrr) rrrr terminatedr rr! env_truncatedenv_terminateds r#convert_to_done_step_apir,Rs| <A< A%%%>J; ze E !J/8/K^+,'i   t $ $7:*83^!NN2?2VDVD./ 8  j4   t $ $}}RVVI.z0BCC/1~~r~~j90+, j4  xy}DzEyFG c<U(a [X5$[X5$)a9Function to transform step returns to the API specified by ``output_truncation_bool``. .. py:currentmodule:: gymnasium.Env Done (old) step API refers to :meth:`step` method returning ``(observation, reward, done, info)`` Terminated Truncated (new) step API refers to :meth:`step` method returning ``(observation, reward, terminated, truncated, info)`` (Refer to docs for details on the API change) Args: step_returns (tuple): Items returned by :meth:`step`. Can be ``(obs, rew, done, info)`` or ``(obs, rew, terminated, truncated, info)`` output_truncation_bool (bool): Whether the output should return two booleans (new API) or one (old) (``True`` by default) is_vector_env (bool): Whether the ``step_returns`` are from a vector environment Returns: step_returns (tuple): Depending on ``output_truncation_bool``, it can return ``(obs, rew, done, info)`` or ``(obs, rew, terminated, truncated, info)`` Example: This function can be used to ensure compatibility in step interfaces with conflicting API. E.g. if env is written in old API, wrapper is written in new API, and the final step output is desired to be in old API. >>> import gymnasium as gym >>> env = gym.make("CartPole-v0") >>> _, _ = env.reset() >>> obs, reward, done, info = step_api_compatibility(env.step(0), output_truncation_bool=False) >>> obs, reward, terminated, truncated, info = step_api_compatibility(env.step(0), output_truncation_bool=True) >>> vec_env = gym.make_vec("CartPole-v0", vectorization_mode="sync") >>> _, _ = vec_env.reset() >>> obs, rewards, dones, infos = step_api_compatibility(vec_env.step([0]), is_vector_env=True, output_truncation_bool=False) >>> obs, rewards, terminations, truncations, infos = step_api_compatibility(vec_env.step([0]), is_vector_env=True, output_truncation_bool=True) )r$r,)routput_truncation_boolrs r#step_api_compatibilityr0sJ7 TT' DDr-)F)rz*DoneStepType | TerminatedTruncatedStepTypereturnTerminatedTruncatedStepType)r*TerminatedTruncatedStepType | DoneStepTyperrr1 DoneStepType)TF)rr3r/rrrr1r3)__doc__ __future__rtypingrrnumpyrgymnasium.corertuplendarrayrrrr4r2r$r,r0r-r#r=s>R"'" '2::  - #$ $  $* $ '2::  - #$ $  $  $* MR3<3 3p 6<666v$((E<(E (E(E0 (Er-