hr'XSrSSKJr SSKrSSKJr SSKJr SSK J r SSK J r SSK rSSKJr SS KJrJrJrJrJrJrJrJrJrJrJrJr /S Qr\S \4SS jj5r\RA\5\RA\5\RA\5\RA\5S \4S S jj5555r!\RA\5S \4S!Sjj5r"\RA\5S \4S"Sjj5r#\RA\5S \4S#Sjj5r$\RA\5S \4S$Sjj5r%\RA\5\RA\5S \4S%Sjj55r&\S&S'Sjj5r'\'RA\5\'RA\5\'RA\5\'RA\5S&S Sjj5555r(\'RA\5S&S!Sjj5r)\'RA\5S&S"Sjj5r*\'RA\5S&S(Sjj5r+\'RA\5S&S)Sjj5r,\S*Sj5r-\-RA\5\-RA\5\-RA\5\-RA\5S+Sj5555r.\-RA\5S,Sj5r/\-RA\5S-Sj5r0\-RA\5S.Sj5r1\-RA\5S/Sj5r2g)0zLUtility functions for vector environments to share memory between processes.) annotationsN)c_bool)singledispatch)SynchronizedArray)Any)CustomSpaceError) BoxDictDiscreteGraph MultiBinary MultiDiscreteOneOfSequenceSpaceTextTupleflatten)create_shared_memoryread_from_shared_memorywrite_to_shared_memoryc[U[5(a$[S[U5S[U5S35e[ S[U5SU35e)aCreate a shared memory object, to be shared across processes. This eventually contains the observations from the vectorized environment. Args: space: Observation space of a single environment in the vectorized environment. n: Number of environments in the vectorized environment (i.e. the number of processes). ctx: The multiprocess module Returns: shared_memory for the shared object across processes. Raises: CustomSpaceError: Space is not a valid :class:`gymnasium.Space` instance Space of type `zH` doesn't have an registered `create_shared_memory` function. Register `z+` for `create_shared_memory` to support it.zVThe space provided to `create_shared_memory` is not a gymnasium Space instance, type: ,  isinstancerrtype TypeErrorspacenctxs ^/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/vector/utils/shared_memory.pyrr!so&%d5k]*rswx}s~r@k l  deijoepdqqstysz {  c URceURRnUS;a[nURX1[ [ R "UR55-5$)N?)dtypecharrArrayintnpprodshape)r!r"r#r(s r$_create_base_shared_memoryr/>sU ;; "" " KK  E | 99UBGGEKK$8 99 ::r%cF^^[UU4SjUR55$)Nc3:># UHn[UTTS9v M g7fr"r#Nr.0subspacer#r"s r$ ._create_tuple_shared_memory..NsEQX4\)tuplespacesr s ``r$_create_tuple_shared_memoryr=Ls! EJ\\ r%c ~URR5VVs0sHup4U[XAUS9_M snn$s snnf)Nr3)r<itemsr)r!r"r#keyr7s r$_create_dict_shared_memoryrASsE %||113 3OS !(S 993  s9cUR[R"[R5RXR -5$N)r*r,r(int32r) max_lengthr s r$_create_text_shared_memoryrF[s/ 99RXXbhh',,a2B2B.B CCr%c^^TR[R"[R5RT54[ UU4SjUR 55-$)Nc3:># UHn[UTTS9v M g7fr2r4r5s r$r8._create_oneof_shared_memory..bs<EQX4\r:)r*r,r(int64r)r;r<r s ``r$_create_oneof_shared_memoryrK`sI IIbhhrxx(--q 1 3e<EJ\\<7 r%c [SUS35e)NzAs zw has a dynamic shape so its not possible to make a static shared memory. For `AsyncVectorEnv`, disable `shared_memory`.)rr s r$_create_dynamic_shared_memoryrMgs$  eWL M r%c[U[5(a$[S[U5S[U5S35e[ S[U5SU35e)a<Read the batch of observations from shared memory as a numpy array. ..notes:: The numpy array objects returned by `read_from_shared_memory` shares the memory of `shared_memory`. Any changes to `shared_memory` are forwarded to `observations`, and vice-versa. To avoid any side-effect, use `np.copy`. Args: space: Observation space of a single environment in the vectorized environment. shared_memory: Shared object across processes. This contains the observations from the vectorized environment. This object is created with `create_shared_memory`. n: Number of environments in the vectorized environment (i.e. the number of processes). Returns: Batch of observations as a (possibly nested) numpy array. Raises: CustomSpaceError: Space is not a valid :class:`gymnasium.Space` instance rzK` doesn't have an registered `read_from_shared_memory` function. Register `z.` for `read_from_shared_memory` to support it.zYThe space provided to `read_from_shared_memory` is not a gymnasium Space instance, type: rrr! shared_memoryr"s r$rros{.%d5k]*uvz|AwBvCCq r  ghlmrhsgttvw|v} ~  r%c[R"UR5URS9R U4UR -5$)Nr()r, frombufferget_objr(reshaper.rOs r$_read_base_from_shared_memoryrVs= ==..0 D L L u{{ r%cT^[U4Sj[XR555$)Nc3<># UHup[X!TS9v M g7fr"Nrr6memoryr7r"s r$r81_read_tuple_from_shared_memory..s""B V A6"B)r;zipr<rOs `r$_read_tuple_from_shared_memoryras& "%m\\"B r%c URR5VVs0sHup4U[XAUUS9_M snn$s snnf)NrZ)r<r?r)r!rPr"r@r7s r$_read_dict_from_shared_memoryrcsJ %||113 3OS $XS/AQ GG3  s<c^[R"UR5[RS9R UTR 45n[ U4SjU55$)NrRc 3># UHOnSRUVs/sH-nU[TR5:dMTRUPM/ sn5v MQ gs snf7f)N)joinlen character_setcharacter_list)r6valuesvalr!s r$r80_read_text_from_shared_memory..sl F " !CU0011*$$S)!     sAA A A)r,rSrTrDrUrEr;)r!rPr"datas` r$_read_text_from_shared_memoryrosY ==..0 A I I E   D     r%c^^[R"USR5[RS9n[ U4Sj[ USSUR 555m[ U4Sj[U555$)NrrRc3<># UHup[X!TS9v M g7frYr[r\s r$r82_read_one_of_from_shared_memory..s""F V A6"Fr_rc3<># UHupUTUU4v M g7frC)r6index sample_indexsubspace_sampless r$r8rrs+#< E ' 5e<=#$Lr%clURR5HupE[XQX$X45 M grC)r<r?r)r!rurkrPr@r7s r$_write_dict_to_shared_memoryrs.++- x ]=OP.r%cURn[R"UR5[RS9n[R "XQU-US-U-[ X55 gr~)rEr,rSrTrDrr)r!rurkrPrrs r$_write_text_to_shared_memoryrsS   D-- 5 5 7rxxHKIIDLEAI#56r%cUupE[R"USR5[RS9n[R"XaUS-U5 [ UR UXUSU-5 g)NrrRr)r,rSrTrJrrr<)r!rurkrP subspace_idx space_valuers r$_write_oneof_to_shared_memoryrsi!'L-- a 0 8 8 :"((KKIIk%!),l; \"E a,FV8Wr%)r!z Space[Any]r"r+return4dict[str, Any] | tuple[Any, ...] | SynchronizedArray)r!,Box | Discrete | MultiDiscrete | MultiBinaryr"r+)r!rr"r+)r!r r"r+)r!rr"r+)r!rr"r+)r!zGraph | Sequencer"r+)r)r!rrPz dict | tuple | SynchronizedArrayr"r+rz-dict[str, Any] | tuple[Any, ...] | np.ndarray)r!rr"r+rztuple[str, ...])r!rr"r+rtuple[Any, ...])r!rrur+r|z np.ndarrayrPr)r!rrur+)r!rrur+rkr)r!r rur+rkzdict[str, Any])r!rrur+rkstr)r!rrur+rkztuple[int, Any])3__doc__ __future__rmultiprocessingmpctypesr functoolsrmultiprocessing.sharedctypesrtypingrnumpyr,gymnasium.errorrgymnasium.spacesr r r r r rrrrrrr__all__rregisterr/r=rArFrKrMrrVrarcrorzrrrrrrrtr%r$rsbR"$:,     X !r   9  8s#x(}-{+BC; 7;