h=SrSSKJr SSKJrJrJr SSKJr SSK J r SSK r SSK JrJr SSKJrJrJr SS KJr SS KJrJrJrJrJr SS KJrJrJr S /r "S S \5r!g)zSImplementation of a synchronous (for loop) vectorization method of any environment.) annotations)CallableIteratorSequence)deepcopy)AnyN)EnvSpace)ActTypeObsType RenderFrame)is_space_dtype_shape_equiv)batch_differing_spaces batch_space concatenatecreate_empty_arrayiterate) ArrayType AutoresetMode VectorEnv SyncVectorEnvc^\rSrSrSrSS\R 4SU4Sjjjr\SSj5r \SSj5r SSS .SS jjr SS jr SS jr SS jrSSjrSSjrSSjrSrU=r$)raYVectorized environment that serially runs multiple environments. Example: >>> import gymnasium as gym >>> envs = gym.make_vec("Pendulum-v1", num_envs=2, vectorization_mode="sync") >>> envs SyncVectorEnv(Pendulum-v1, num_envs=2) >>> envs = gym.vector.SyncVectorEnv([ ... lambda: gym.make("Pendulum-v1", g=9.81), ... lambda: gym.make("Pendulum-v1", g=1.62) ... ]) >>> envs SyncVectorEnv(num_envs=2) >>> obs, infos = envs.reset(seed=42) >>> obs array([[-0.14995256, 0.9886932 , -0.12224312], [ 0.5760367 , 0.8174238 , -0.91244936]], dtype=float32) >>> infos {} >>> _ = envs.action_space.seed(42) >>> actions = envs.action_space.sample() >>> obs, rewards, terminates, truncates, infos = envs.step(actions) >>> obs array([[-0.1878752 , 0.98219293, 0.7695615 ], [ 0.6102389 , 0.79221743, -0.8498053 ]], dtype=float32) >>> rewards array([-2.96562607, -0.99902063]) >>> terminates array([False, False]) >>> truncates array([False, False]) >>> infos {} >>> envs.close() Tsamec>[TU]5 XlX lX0l[ U[ 5(aUO [ U5UlUVs/sH oU"5PM snUl[UR5Ul URSRUl URURS'URSRUl URSRUl[URUR5Ul [ U[ 5(aS[U5S:XaD[ US["5(de[ US["5(deUuUlUlOUS:XaDURSR$Ul[UR&UR5UlOgUS:XaSURSR$Ul[)URVs/sHofR$PM sn5UlO[+SU35eURHnUS:Xa=UR$UR&:Xd"SUR&S UR$S 35eOF[-UR$UR&5(d!S UR&S UR$35eURUR:XaMS URSUR35e [/UR5Vs/sHnSPM snUl[3UR&UR[4R6S9Ul[4R6"UR4[4R:S9Ul[4R6"UR4[4R>S9Ul [4R6"UR4[4R>S9Ul![4R6"UR4[4R>S9Ul"gs snfs snfs snf)aVectorized environment that serially runs multiple environments. Args: env_fns: iterable of callable functions that create the environments. copy: If ``True``, then the :meth:`reset` and :meth:`step` methods return a copy of the observations. observation_mode: Defines how environment observation spaces should be batched. 'same' defines that there should be ``n`` copies of identical spaces. 'different' defines that there can be multiple observation spaces with the same length but different high/low values batched together. Passing a ``Space`` object allows the user to set some custom observation space mode not covered by 'same' or 'different.' autoreset_mode: The Autoreset Mode used, see https://farama.org/Vector-Autoreset-Mode for more information. Raises: RuntimeError: If the observation space of some sub-environment does not match observation_space (or, by default, the observation space of the first sub-environment). rautoreset_moder differentzwInvalid `observation_mode`, expected: 'same' or 'different' or tuple of single and batch observation space, actual got zSyncVectorEnv(..., observation_mode='same') however the sub-environments observation spaces are not equivalent. single_observation_space=z$, sub-environment observation_space=zE. If this is intentional, use `observation_mode='different'` instead.zSyncVectorEnv(..., observation_mode='different' or custom space) however the sub-environments observation spaces do not share a common shape and dtype, single_observation_space=z$, sub-environment observation space=zRSub-environment action space doesn't make the `single_action_space`, action_space=z, single_action_space=N)nfndtype)#super__init__env_fnscopyobservation_mode isinstancerrenvslennum_envsmetadata render_mode action_spacesingle_action_spacertupler observation_spacesingle_observation_spacer ValueErrorrrange_env_obsrnpzeros _observationsfloat64_rewardsbool_ _terminations _truncations_autoreset_envs) selfr&r'r(renv_fnenv_ __class__s Z/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/vector/sync_vector_env.pyr%SyncVectorEnv.__init__@s*   0.-88 ~. -44G&VXG4 DII  ! -- *.*=*= &'99Q<33#'99Q<#<#< '(@(@$--P & . .37G3HA3M.q1599 99.q1599 99DT AD "D$A6)04 ! 0N0N-)4114==*&"[004 ! 0N0N-)?6:ii@is**i@*&!NO_N`a 99C6)))T-J-JJ_`d`}`}_~~bcfcxcxbyy~J2))4+H+HbGHLHeHeGffJKNK`K`Jabb   D$<$<< gcdgdtdtcuvLMQMeMeLfg g<(-T]]';<';!';< /  ) )T]]rxx $--!1D XXt}}&6bhhGHHdmm%5RXXF!xx(8Iq52A.=s P1P6 P;c$URS5$)z8Returns a tuple of np random seeds for the wrapped envs.np_random_seedget_attrr@s rErHSyncVectorEnv.np_random_seeds}}-..c$URS5$)zKReturns a tuple of the numpy random number generators for the wrapped envs. np_randomrIrKs rErOSyncVectorEnv.np_randoms}}[))rMNseedoptionscUc&[UR5Vs/sHnSPM nnO<[U[5(a'[UR5Vs/sHoAU-PM nn[ U5UR:Xd!SURS[ U5S35eUGbsSU;GalUR S5n[U[ R5(dS[U535eURUR4:Xd!SURSUR35eUR[ R:XdS UR35e[ R"U5(d S U35eS URU'S URU'S URU'0n[![#UR$X55HCunupxn U (dMUR'XS 9uUR(U'n UR+XjU5nME O[ R,"UR4[ RS 9Ul [ R,"UR4[ RS 9Ul[ R,"UR4[ RS 9Ul0n[![#UR$U55H9unupxUR'XS 9uUR(U'n UR+XjU5nM; [/UR0UR(UR25UlUR4(a[7UR25U4$UR2U4$s snfs snf)aResets each of the sub-environments and concatenate the results together. Args: seed: Seeds used to reset the sub-environments, either * ``None`` - random seeds for all environment * ``int`` - ``[seed, seed+1, ..., seed+n]`` * List of ints - ``[1, 2, 3, ..., n]`` options: Option information used for each sub-environment Returns: Concatenated observations and info from each sub-environment Nz=If seeds are passed as a list the length must match num_envs=z but got length=. reset_maskz9`options['reset_mask': mask]` must be a numpy array, got z0`options['reset_mask': mask]` must have shape `(z ,)`, got z>`options['reset_mask': mask]` must have `dtype=np.bool_`, got zK`options['reset_mask': mask]` must contain a boolean array, got reset_mask=FrQr")r5r,r)intr+popr7ndarraytypeshaper#r<anyr=r>r? enumeratezipr*resetr6 _add_infor8rr3r9r'r) r@rRrSrCirVinfosrB single_seedenv_maskenv_infos rEr_SyncVectorEnv.resetsP$ <"' "67"6QD"6D7D c " "&+DMM&:;&:1H&:D; I & w J4==/Yijmnrjsittu v w &  <7#: \2JBJJ ^J4PZK[J\] ^## ( mA$--PYZdZjZjYkl m  BHH, cOPZP`P`Oab c,66 j\]g\hi j.3D  z *,1D  j )/4D  ,E3<DIIt04//Ch814(2;2.DMM!$h!NN5A>E4"$4==*:"((!KD  "$--)9 JD #%88T]],P>PRWWWk8;s M4M9c[URU5n0n[[XRSS95GHunupEUR [ R:XaURU(a\URUR5uURU'nSURU'SURU'SURU'GOURURU5uURU'URU'URU'URU'nGOUR [ R:XaURU(aSUR<35eURURU5uURU'URU'URU'URU'nGOUR [ R :XaURURU5uURU'URU'URU'URU'nURU(dURU(aPUR#UURUUS.U5nURUR5uURU'nO[%SUR 35eUR#X&U5nGM ['UR(URUR*5Ul[,R."URUR5UlUR0(a[3UR*5O UR*[,R0"UR5[,R0"UR5[,R0"UR5U4$)zySteps through each of the environments returning the batched results. Returns: The batched environment step results T)strictgFzself._autoreset_envs=) final_obs final_infozUnexpected autoreset mode, )rr/r]r^r*rr NEXT_STEPr?r_r6r;r=r>stepDISABLED SAME_STEPr`r4rr3r9r7 logical_orr'r)r@actionsrbraactionrCres rErlSyncVectorEnv.steps7$++W5'GYYt(LMNA{""m&=&==''*1511C1C1E.DMM!$h'*DMM!$,1D&&q)+0D%%a( ! ))&1  a( a(**1-))!, $$ (>(>>//2N7Lt7K7K6M4NN2IIaL%%f- MM!$MM!$&&q)%%a($$ (?(??IIaL%%f- MM!$MM!$&&q)%%a(%%a(D,=,=a,@ NN&*mmA&6hOE 2611C1C1E.DMM!$h #>t?R?R>S!TUUNN5A6E]Nb)  ) )4==$:L:L  "}}T-?-?ARARS-1IIHT'' (4;M;M GGDMM " GGD&& ' GGD%% &    rMc:[SUR55$)z2Returns the rendered frames from the environments.c3@# UHoR5v M g7f)N)render).0rBs rE 'SyncVectorEnv.render..:s7YcZZ\\Ys)r1r*rKs rEruSyncVectorEnv.render8s7TYY777rMc/nURHNnURU5n[U5(aURU"U0UD65 M=URU5 MP [ U5$)zCalls a sub-environment method with name and applies args and kwargs. Args: name: The method name *args: The method args **kwargs: The method kwargs Returns: Tuple of results )r*get_wrapper_attrcallableappendr1)r@nameargskwargsresultsrBfunctions rEcallSyncVectorEnv.call<s`99C++D1H!!x889x( W~rMc$URU5$)zGet a property from each parallel environment. Args: name (str): Name of the property to get from each individual environment. Returns: The property with name )r)r@r~s rErJSyncVectorEnv.get_attrRsyyrMch[U[[45(d$[UR5Vs/sHo2PM nn[ U5UR:wa%[ S[ U5SURS35e[URU5HupEURX5 M gs snf)aSets an attribute of the sub-environments. Args: name: The property name to change values: Values of the property to be set to. If ``values`` is a list or tuple, then it corresponds to the values for each individual environment, otherwise, a single value is set for all environments. Raises: ValueError: Values must be a list or tuple with length equal to the number of environments. zUValues must be a list or tuple with length equal to the number of environments. Got `z ` values for z environments.N) r)listr1r5r,r+r4r^r*set_wrapper_attr)r@r~valuesrCrBvalues rEset_attrSyncVectorEnv.set_attr]s&4-00&+DMM&:;&:f&:F; v;$-- 'F }M$--P  dii0JC   -1r/rr'r&r*r-r,r(r2r.r0r3)r&z9Iterator[Callable[[], Env]] | Sequence[Callable[[], Env]]r'boolr(z str | Spacerzstr | AutoresetMode)returnztuple[int, ...])rztuple[np.random.Generator, ...])rRzint | list[int | None] | NonerSzdict[str, Any] | Nonerztuple[ObsType, dict[str, Any]])rpr rz?tuple[ObsType, ArrayType, ArrayType, ArrayType, dict[str, Any]])rztuple[RenderFrame, ...] | None)r~strrrrrrtuple[Any, ...])r~rrr)r~rrz!list[Any] | tuple[Any, ...] | Any)rr)__name__ __module__ __qualname____firstlineno____doc__rrkr%propertyrHrOr_rlrurrJrr__static_attributes__ __classcell__)rDs@rErrs"N(..;.E.E YJJYJYJ& YJ , YJYJv//**/3)- HX,HX' HX ( HXTG G HG R8, .0//rM)"r __future__rcollections.abcrrrr'rtypingrnumpyr7 gymnasiumr r gymnasium.corer r r gymnasium.spaces.utilsrgymnasium.vector.utilsrrrrrgymnasium.vector.vector_envrrr__all__rrMrErsQY"88 88=LK  ]/I]/rM